Neural Tangents

github.comtool

High-level API for specifying neural networks of both finite and infinite width.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Jraph github.com

   tool category #python-data-science link ↗

   A Graph Neural Network Library in Jax.

   
2. ▲
   Jraph github.com

   tool category #jax link ↗

   Lightweight graph neural network library.

3. ▲
   FLAX github.com

   tool category #python-data-science link ↗

   A neural network library for JAX that is designed for flexibility.

4. ▲
   dendritex github.com

   tool category #jax link ↗

   Dendritic Modeling in JAX.

   
5. ▲
   Equivariant MLP github.com

   tool category #jax link ↗

   Construct equivariant neural network layers.

6. ▲
   GPJax github.com

   tool category #jax link ↗

   Gaussian processes in JAX.

   

JAX › Libraries: “High-level API for specifying neural networks of both finite and infinite width.”