GitHub repository: ml-gde/jax-layers
Primary language: Python
Description: A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax