Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
Why do you think that https://github.com/google/trax is a good alternative to jax-resnet
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
Why do you think that https://github.com/google/trax is a good alternative to jax-resnet