Skip to content

rwightman/efficientnet-jax

Error
Looks like something went wrong!

About

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

Topics

flax mixnet tpu jax mobilenetv2 mobilenetv3 efficientnet objax flax-linen

Resources

Readme

License

Apache-2.0 license
Activity

Stars

128 stars

Watchers

5 watching

Forks

14 forks
Report repository

Releases 1

pretrained weights ported from timm (PyTorch) and original tensorflow weights Latest
Oct 27, 2020

Packages

No packages published

Contributors 2

  •  
  •  

Languages