notebook.community
Edit and run
In [0]: https://github.com/google/flax
https://github.com/google/flax