/jax-resnet

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

Primary LanguagePythonMIT LicenseMIT

No issues in this repository yet.