ml-gde/jaxgarden

A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax