If you're a JAX user and you need pretrained models, give it a try. All Keras models act as native JAX functions (with the JAX backend).

"JAX doesn't have pretrained models" is a myth -- it has thousands of pretrained Keras 3 models.