Sanchit Gandhi ee27b3d7df Replace all deprecated jax.ops operations with jnp's at (#16078) ...

* Replace all deprecated `jax.ops` operations with jnp's `at`

* np to jnp scores

* suggested changes

2022-03-16 09:08:55 +00:00

..

adding_a_new_example_script

Happy New Year! (#15094)

2022-01-10 12:05:57 -05:00

adding_a_new_model

Replace all deprecated jax.ops operations with jnp's at (#16078)

2022-03-16 09:08:55 +00:00