flax

Python ★ 0 updated 2y 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.