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 |
|
Tom Hennigan
|
ca3f9e9599
|
Internal change
PiperOrigin-RevId: 419592537
|
2022-02-16 16:09:34 +00:00 |
|
Loren Maggiore
|
68473a2243
|
Update NFNet optim to reflect Haiku default of dict instead of FlatMapping.
PiperOrigin-RevId: 418005668
|
2022-02-16 16:09:15 +00:00 |
|
Loren Maggiore
|
3146ff9f5d
|
Fix dataset options usage.
PiperOrigin-RevId: 415295087
|
2021-12-10 23:15:11 +00:00 |
|
Loren Maggiore
|
d55816beb3
|
Formatting fixes and internal changes.
PiperOrigin-RevId: 414656804
|
2021-12-09 14:00:10 +00:00 |
|
Peter Hawkins
|
a7d75013c9
|
Remove users of jax.api.* symbols, in preparation for removing the deprecated jax.api name.
In most cases, use the public jax.* name instead.
PiperOrigin-RevId: 395812511
|
2021-09-17 17:49:29 +01:00 |
|
Jiri Simsa
|
0015af5805
|
[tf.data] Moving autotuning options from tf.data.Options.experimental_optimization.autotune* to a newly created tf.data.Options.autotune.* and removing support for tf.data.Options.experimental_optimization.autotune_buffers.
PiperOrigin-RevId: 389487692
|
2021-08-09 17:19:38 +01:00 |
|
Alistair Muldal
|
9c9e41e1fc
|
Enable Travis tests for nfnets
PiperOrigin-RevId: 368236627
|
2021-04-13 21:03:28 +01:00 |
|
Andy Brock
|
ba761289c1
|
Upload links to pre-trained NFNet weights and add utility functions for loading them, as well as demo colab notebook.
PiperOrigin-RevId: 357692801
|
2021-02-17 17:42:56 +00:00 |
|
Andy Brock
|
68a1754d29
|
Initial commit for NFNets and AGC
PiperOrigin-RevId: 357022514
|
2021-02-12 08:48:41 +00:00 |
|
Louise Deason
|
abc9ba0635
|
Initial release of "nfnets".
PiperOrigin-RevId: 356975781
|
2021-02-11 16:42:59 +00:00 |
|