gitmyhub

flax

Jupyter Notebook ★ 0 updated 2mo 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.