note on setting OQUPY_BACKEND variable

tempoCollaboration · Nov 10, 2024 · 1813d8b · 1813d8b
1 parent 6cf9216
commit 1813d8b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/DEVELOPMENT.md b/DEVELOPMENT.md
@@ -24,6 +24,8 @@ To enable experimental features switch to the `dev/jax` branch and use
 from oqupy.backends import enable_jax_features
 enable_jax_features()
 ```
+Alternatively, the `OQUPY_BACKEND` environmental variable may be set to `jax` to
+initialize the jax backend by default.
 
 ### Contributing Guidelines
 

diff --git a/docs/pages/gpu_features.rst b/docs/pages/gpu_features.rst
@@ -25,6 +25,9 @@ To enable experimental features, switch to the ``dev/jax`` branch and use
    from oqupy.backends import enable_jax_features
    enable_jax_features()
 
+Alternatively, the `OQUPY_BACKEND` environmental variable may be set to `jax` to
+initialize the jax backend by default.
+
 Contributing Guidelines
 ~~~~~~~~~~~~~~~~~~~~~~~
 

diff --git a/examples/simple_dynamics_jax.py b/examples/simple_dynamics_jax.py
@@ -8,8 +8,8 @@
 import oqupy
 from oqupy.backends import enable_jax_features
 # import NumPy from numerical_backend
-from oqupy.backends.numerical_backend import np
-enable_jax_features()
+#from oqupy.backends.numerical_backend import np
+#enable_jax_features()
 
 import matplotlib.pyplot as plt
 sigma_x = oqupy.operators.sigma("x")
@@ -39,5 +39,5 @@
 plt.plot(t, s_z, label=r'$\alpha=0.3$')
 plt.xlabel(r'$t\,\Omega$')
 plt.ylabel(r'$\langle\sigma_z\rangle$')
-#plt.savefig('simple_dynamics.png')
-plt.show()
+plt.savefig('simple_dynamics_jax.png')
+#plt.show()