congyingxia/flax
Flax is a neural network library for JAX that is designed for flexibility.
PythonApache-2.0
No issues in this repository yet.
Flax is a neural network library for JAX that is designed for flexibility.
PythonApache-2.0
No issues in this repository yet.