flax. Flax is a neural network library for JAX that is designed for flexibility.

github.com/juntang-zhuang/flax

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.