jimm (pypi): JAX Image Models
Keywords: jax|flax|linen|pretrained|models|efficientnet|mobilenetv3|mnasnet