Use JIT compiler for `StatevectorBackend`
Before submitting, please check the following:
- Make sure you have tests for the new code and that test passes (run
tox) - format added code by
black -l 120 <filename> - If applicable, add a line to the [unreleased] part of CHANGELOG.md, following keep-a-changelog.
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.
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.
closing becuase of inactivity. happy to reopen anytime if there's interest!