flax

Python ★ 1 updated 3y 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.