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

Faster optimization for heterodyne likelihood using Jax #7

Closed

kazewong opened this issue May 11, 2023 · 0 comments · Fixed by #9

Assignees

Owner

kazewong commented May 11, 2023

Currently, the optimization is done through scipy differential evolution by calling the jax kernel probably sequentially, which is pretty slow.

Ideally we want something completely Jaxxed, such as a vmap lbfgs to solve for the optimum

kazewong self-assigned this

kazewong linked a pull request

that will close this issue

7 faster optimization for heterodyne likelihood using jax #9

Merged

kazewong closed this as completed in #9

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