ml-gde / jax-layers

A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax
β˜†11Updated last week

Alternatives and similar repositories for jax-layers:

Users that are interested in jax-layers are comparing it to the libraries listed below