Package Usage: pypi: flax
Flax: A neural network library for JAX designed for flexibility
46 versions
Latest release: about 1 year ago
198 dependent packages
882,615 downloads last month
View more package details: https://packages.ecosystem.code.gouv.fr/registries/pypi.org/packages/flax