[BUG] BasisState does not work with jax.jit
#999
| Job | Run time |
|---|---|
| 7s | |
| 7s |
BasisState does not work with jax.jit
#999
| Job | Run time |
|---|---|
| 7s | |
| 7s |