4 Commits

Author	SHA1	Message	Date
Suraj Patil	93d3fd8645	remove jax.ops.index (#16220)	2022-03-17 17:51:43 +01:00
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
Nicholas Broad	69e16abf98	Switch from using sum for flattening lists of lists in group_texts (#14472) ... * remove sum for list flattening * change to chain(*) * make chain object a list * delete empty lines per sgugger's suggestions Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Co-authored-by: Nicholas Broad <nicholas@nmbroad.com> Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>	2021-11-22 16:17:26 -05:00
Suraj Patil	8ef3f36561	fix typos (#12757)	2021-07-16 16:44:59 +05:30