Automatically jit certain functions in jaxquantum

Open Phionx opened this issue 3 months ago • 1 comments

Previously, we were worried about vmapping jitted functions, but it seems that is not an issue, as long as there is a final jit.

Oct 09 '25 18:10 Phionx

In solvers.py we can jit solve() with a static argument for the ODE function f. That should cover both sesolve and mesolve I think.

What else?

Oct 31 '25 21:10 shoumikdc