afqueiruga/flax
Flax is a neural network ecosystem for JAX that is designed for flexibility.
PythonApache-2.0
Stargazers
No one’s star this repository yet.
Flax is a neural network ecosystem for JAX that is designed for flexibility.
PythonApache-2.0
No one’s star this repository yet.