Sanchit Gandhi
|
e6f221c8d4
|
[JAX] Replace all jax.tree_* calls with jax.tree_util.tree_* (#18361)
* [JAX] Replace all jax.tree_* calls with jax.tree_util.tree_*
* fix double tree_util
|
2022-09-09 15:18:56 +02:00 |
|
Yulv-git
|
95113d1365
|
Fix some typos. (#17560)
* Fix some typos.
Signed-off-by: Yulv-git <yulvchi@qq.com>
* Fix typo.
Signed-off-by: Yulv-git <yulvchi@qq.com>
* make fixup.
|
2022-07-11 05:00:13 -04:00 |
|
Sylvain Gugger
|
afe5d42d8d
|
Black preview (#17217)
* Black preview
* Fixup too!
* Fix check copies
* Use the same version as the CI
* Bump black
|
2022-05-12 16:25:55 -04:00 |
|
Stas Bekman
|
a73281e3e4
|
[examples] max samples can't be bigger than the len of dataset (#16501)
* [examples] max samples can't be bigger than then len of dataset
* do tf and flax
|
2022-03-30 12:33:16 -07:00 |
|
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 |
|