Ad Space
flax
Flax: A neural network library for JAX designed for flexibility
v0.12.5
Exclude CI/CDPRO
Download Trend
Version Downloads
Python Versions
Operating Systems
Installers
Ad Space
Badge
Markdown
[](/packages/flax)
HTML
<a href="/packages/flax"><img src="/api/badges/flax?period=month" alt="PyPI Stats"></a>