Commit Graph

3 Commits

Author SHA1 Message Date
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
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
Louise Deason abc9ba0635 Initial release of "nfnets".
PiperOrigin-RevId: 356975781
2021-02-11 16:42:59 +00:00