colab_jaxbook

Colab Notebooks for JAX/Flax/Optax ML Book

Disclaimer: This is not an official Google product