Suggest an 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/matthias-wright/flaxmodels is a good alternative to jax-resnet

A URL to the alternative repo (e.g. GitHub, GitLab)

Here you can share your experience with the project you are suggesting or its comparison with jax-resnet. Optional.

A valid email to send you a verification link when necessary or log in.