Stephen Wang

Results 1 comments of Stephen Wang

downgrade jax to 0.2.21 jax.ops.index_add is deprecated in 0.2.22 https://gitee.com/mirrors/JAX/blob/main/CHANGELOG.md#jax-0222-oct-12-2021