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

[Bug] Conjugate gradients algorithm Solve with JAX uses a deprecated function #111

Open
adam-hartshorne opened this issue Dec 4, 2024 · 1 comment
Labels
bug Something isn't working

Comments

@adam-hartshorne
Copy link

🐛 Bug

If one attempts to use the CG algorithm to solve for an inverse, it will call host_callback.id_tap(construct_info, None) inside of jax_tqdm.py. However, jax.experimental.host_callback has been deprecated since March 2024 and is now no longer supported. See jax-ml/jax#20385

@adam-hartshorne adam-hartshorne added the bug Something isn't working label Dec 4, 2024
@AndPotap
Copy link
Collaborator

@adam-hartshorne, I'm looking into this.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

2 participants