2 results found Sort:

rwightman
efficientnet-jax rwightman
14
125
apache-2.0
6
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
jax tpu flax objax mixnet flax-linen mobilenetv2 mobilenetv3 efficientnet
Created 2020-09-12
50 commits to master branch, last one 5 months ago
alvarobartt
safejax alvarobartt
4
41
mit
2
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
jax flax haiku objax safejax huggingface safetensors serialization jax-serialization flax-serialization haiku-serialization objax-serialization
Created 2022-12-21
151 commits to main branch, last one about a year ago