Merge pull request #55 from rochisha0/add-settings

Add jax.numpy backend

src/qutip_jax/__init__.py

@@ -34,6 +34,7 @@
 from .permute import *
 from .reshape import *
 from . import norm
+from .settings import *
 from .measurements import *
 from .properties import *
 from .linalg import *

src/qutip_jax/settings.py

@@ -0,0 +1,7 @@
+import jax.numpy as jnp
+from qutip import settings
+
+__all__ = ["use_jax_backend"]
+
+def use_jax_backend():
+    settings.core['numpy_backend'] = jnp