penzai

Python ★ 1.9k updated 1y ago

A JAX research toolkit for building, editing, and visualizing neural networks.

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