EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
Primary LanguagePythonApache License 2.0Apache-2.0