EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
-
Updated
Jan 4, 2024 - Python
#
objax
Here are 3 public repositories matching this topic...
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
serialization
flax
haiku
jax
huggingface
objax
safetensors
jax-serialization
flax-serialization
objax-serialization
haiku-serialization
safejax
-
Updated
May 31, 2024 - Python
Programming assignment for UofT ECE421, Fall 2020: CNN Feature Visualization using jax and objax.
-
Updated
Aug 18, 2021 - Jupyter Notebook
Improve this page
Add a description, image, and links to the objax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the objax topic, visit your repo's landing page and select "manage topics."