objax

There are 3 repositories under objax topic.

rwightman/efficientnet-jax
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
Language:Python127 6 314
alvarobartt/safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
Language:Python42 2 135
mamaj/cnn-featurevis-ece421
Programming assignment for UofT ECE421, Fall 2020: CNN Feature Visualization using jax and objax.
Language:Jupyter Notebook1 1 11