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

[Engines, Backends] Add JAX Engine and Horqrux Backend #111

Merged
merged 114 commits into from
Dec 11, 2023
Merged

Conversation

dominikandreasseitz
Copy link
Collaborator

@dominikandreasseitz dominikandreasseitz commented Oct 17, 2023

Addresses #93

  1. This MR adds an Engine to qadence which is uses to instantiate a DifferentiableBackend which can be either JAX or torch.
  2. Each backend gets a engine field which is uses by backend_factory to import either a torch or jax differentiable backend.
  3. Add a new backend called 'horqrux' https://github.com/pasqal-io/horqrux, a differntiable state vector simulator in JAX
  4. Add diff_modes GPSR and AD to horqrux backend
  5. Add low-level tutorial to train qadence circuits in JAX

@dominikandreasseitz dominikandreasseitz marked this pull request as draft October 17, 2023 15:31
@dominikandreasseitz dominikandreasseitz self-assigned this Oct 17, 2023
@dominikandreasseitz dominikandreasseitz added the proto Experimental features label Oct 17, 2023
@dominikandreasseitz dominikandreasseitz added feature New feature or request core-dev backends Improving the backends and removed proto Experimental features core-dev labels Nov 17, 2023
@dominikandreasseitz dominikandreasseitz changed the title [Proto] Add JAX DifferentiableBackend [Backends] Add JAX DifferentiableBackend Nov 17, 2023
qadence/backend.py Show resolved Hide resolved
qadence/backends/api.py Show resolved Hide resolved
qadence/backends/horqrux/backend.py Show resolved Hide resolved
qadence/backends/horqrux/config.py Show resolved Hide resolved
qadence/engines/differentiable_backend.py Show resolved Hide resolved
qadence/types.py Show resolved Hide resolved
qadence/extensions.py Outdated Show resolved Hide resolved
Copy link
Collaborator

@GJBoth GJBoth left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks so much for this :-)

@dominikandreasseitz dominikandreasseitz merged commit 5a7c8dd into main Dec 11, 2023
8 checks passed
@dominikandreasseitz dominikandreasseitz deleted the ds/jax branch December 11, 2023 15:05
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
backends Improving the backends engines feature New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

6 participants