GitHub repository: n2cholas/jax-resnet
Primary language: Python
Description: Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).