google/flax
View on GitHub ↗ ★ 7,365 Jupyter Notebook
Flax is a neural network library for JAX that is designed for flexibility.
6 mentions on Hacker News by 6 people: 1 story, 5 comments · first 2020-04-08 · latest 2025-12-15
20082026
-
▲ 2
Flax: A neural network library and ecosystem for Jax designed for flexibilitystory by abracos · 2025-12-15 · 0 comments