flax

★ 0 updated 6y ago ⑂ fork

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

No plain-English explanation yet — one is being written right now. Check back in a minute.