jaxquantum
jaxquantum copied to clipboard
EQuS
jax + quantum dynamics simulations
The example in the README fails [1] when running the following line: states = jqt.mesolve(g_state_dm, ts, c_ops=c_ops, Ht=Ht) which appears to be because of the ordering of the arguments to...
Added the energy restricted space (enr) feature to `jaxquantum`. The code structure follows the QuTip implementation of the same feature.
For now, this uses JAX BCOO. But now, it should be quite easy to implement other data backends. I still have a few things to add before this PR is...
This is a first, low-effort pass at making it easier to move from qutip to jaxquantum. Some issues already identified: * Add an identity operator constructor with the same name...
Previously, we were worried about vmapping jitted functions, but it seems that is not an issue, as long as there is a final jit.
Inspired by https://www.dynamiqs.org/stable/python_api/plot/wigner_gif.html. Would be very useful for visualization of operations on bosonic modes.
