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.

Sign up for GitHub

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

Jump to bottom

[Proto] Use apply_gate via jax.lax.scan #19

Open

RolandMacDoland opened this issue Apr 26, 2024 · 1 comment

Labels

proto

Collaborator

RolandMacDoland commented Apr 26, 2024

@dominikandreasseitz Description ?

RolandMacDoland assigned dominikandreasseitz

RolandMacDoland added the proto label

RolandMacDoland mentioned this issue

[Proto] Use apply_gate via jax.lax.scan #15

Closed

mlahariya unassigned dominikandreasseitz

Collaborator

mlahariya commented Jan 23, 2025

Jax.lax.scan is the equivalent of doing for loops.

It's recommended to do it in case of tensor manipulations for correct compilation.
https://jax.readthedocs.io/en/latest/_autosummary/jax.lax.scan.html

All for loops should be replaced by Jax.lax.scan.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment