Replies: 1 comment 11 replies
-
Thanks for the question. It's a bit tricky to explain in full. We plan to write an updated version of autodidax explaining the new core implementation, but we don't have that yet. Maybe the best thing to do is to look at how that commit changed code involving new_main, e.g. in ad.py, since the changes needed on quax are likely analogous. |
Beta Was this translation helpful? Give feedback.
11 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
The library
quax
relies (code here) on the functionjax.core.new_main
, which was removed in v0.4.36 (commit).Does anyone know if there's an alternative means to substitute Jax tracers?
Beta Was this translation helpful? Give feedback.
All reactions