Colab Notebooks for JAX/Flax/Optax ML Book
Primary LanguageJupyter NotebookApache License 2.0Apache-2.0