flax
Flax is a neural network library for JAX that is designed for flexibility.
6k stars
86 watching
632 forks
Language: Python
last commit: 4 days ago
Linked from 2 awesome lists
jax
flax Flax is a neural network library for JAX that is designed for flexibility.
krzjoa/awesome-python-data-science
amrzv/awesome-colab-notebooks