parikshit-adhikari / flax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from google/flax
Flax is a neural network library for JAX that is designed for flexibility.
Home Page: https://flax.readthedocs.io
License: Apache License 2.0