/jax-resnet

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

Primary LanguagePythonMIT LicenseMIT

Watchers

No one’s watching this repository yet.