Source code updated to support Jax 0.3.14 #29
shailesh1729
announced in
Announcements
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Recent versions of JAX have dropped support for index_add and index_update functions in
jax.ops
module. The older code was still using them. I have updated the source code for cr-sparse and cr-nimble to work with the recent Jax release 0.3.14. The code from the master branch is now working fine. I will make a new official release in some time.Beta Was this translation helpful? Give feedback.
All reactions