GitHub repository: rwightman/efficientnet-jax
Primary language: Python
Description: EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax