jax-resnet

github.comtool

Implementations and checkpoints for ResNet variants in Flax.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   FLAX github.com

   tool category #python-data-science link ↗

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

2. ▲
   flaxmodels github.com

   tool category #jax link ↗

   Pretrained models for Jax/Flax.

3. ▲
   Accurate Quantized Training github.com

   site category #jax link ↗

   Tools and libraries for running and analyzing neural network quantization experiments in JAX and Flax.

4. ▲
   FedJAX github.com

   tool category #jax link ↗

   Federated learning in JAX, built on Optax and Haiku.

   
5. ▲
   Flax NNX github.com

   site category #jax link ↗

   An evolution on Flax by the same team

6. ▲
   JAX RL github.com

   tool category #jax link ↗

   Implementations of reinforcement learning algorithms.

   

JAX › Libraries > New Libraries: “Implementations and checkpoints for ResNet variants in Flax.”