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
-
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