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
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.
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
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