arg name change in  JS_divergence in linalg.py