You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The JAX team is going to remove optix from the library now that optax exists, can I open a PR changing all imports from jax.experimental.optix to optax?
The text was updated successfully, but these errors were encountered:
Thanks for the suggestion @joaogui1, Matteo (the primary author of optax) already has a patch proposed internally that will get merged quite soon 😄 Will close this issue once that's in.
The JAX team is going to remove optix from the library now that optax exists, can I open a PR changing all imports from
jax.experimental.optix
tooptax
?The text was updated successfully, but these errors were encountered: