 Peter Hawkins
|
82a347438f
|
Fix pytype failures related to teaching pytype about NumPy scalar types.
PiperOrigin-RevId: 519205179
|
2023-06-02 18:03:40 +01:00 |
|
|
Peter Hawkins
|
c051e6a51d
|
Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.
PiperOrigin-RevId: 511294746
|
2023-06-02 18:02:16 +01:00 |
|
 Jake VanderPlas
|
6fcb84268e
|
Use jax.tree_util.tree_map in place of deprecated tree_multimap.
The latter is a simple alias of the former, so this change is a no-op.
PiperOrigin-RevId: 461229165
|
2022-07-24 17:53:28 +01:00 |
|
 Florent Altché
|
8457046b2c
|
Add checkpoints from the ablation study.
PiperOrigin-RevId: 328023346
|
2020-08-26 16:54:56 +01:00 |
|
 Clara Huiyi Hu
|
923ad3cff0
|
This notebook illustrates the CIFAR-10 experiments in the paper:
PiperOrigin-RevId: 326141025
|
2020-08-21 08:16:49 +00:00 |
|