jax-resnet

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

GitHub
103 stars
4 watching
8 forks
Language: Python
last commit: over 2 years ago
Linked from 1 awesome list

computer-visiondeep-learningflaxjaxmachine-learningneural-networksresnet

Screenshot of n2cholas/jax-resnet website

https://pypi.org/project/jax-resnet/

Backlinks from these awesome lists: