flax-linen topic

List flax-linen repositories

efficientnet-jax

123
Stars
14
Forks
Watchers

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

verified publisher icon rwightman