/colab_jaxbook

Colab Notebooks for JAX/Flax/Optax ML Book

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Issues