Open king-p3nguin opened 6 months ago
Currently, I am using numba
, but I am going to change it to jax
because jax
seems to support more functions in numpy
. (functions like np.tensordot
and np.moveaxis
is not supported in numba
)
I will make the installation of jax
optional.
It seems like jaxlib
does not support Python 3.8 on Windows, so some tests involving Jax are skipped. Also, since there are many optional dependent libraries, I added a new requirements-dev.txt
file.
Before submitting, please check the following:
tox
)black -l 120 <filename>
Then, please fill in below:
Context (if applicable):
Description of the change: Use JIT for faster simulation using
StatevectorBackend
.Related issue:
also see that checks (github actions) pass. If lint check keeps failing, try installing black==22.8.0 as behavior seems to vary across versions.