Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Use JIT compiler for StatevectorBackend #110

Draft
wants to merge 31 commits into
base: master
Choose a base branch
from

Conversation

king-p3nguin
Copy link
Contributor

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.

@king-p3nguin
Copy link
Contributor Author

king-p3nguin commented Dec 24, 2023

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.

@king-p3nguin king-p3nguin changed the title Use JIT for StatevectorBackend Use JIT compiler for StatevectorBackend Dec 26, 2023
@king-p3nguin
Copy link
Contributor Author

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant