objax

There are 0 repository under objax topic.

rwightman / efficientnet-jax
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
efficientnet flax flax-linen jax mixnet mobilenetv2 mobilenetv3 objax tpu
Language:Python 123
alvarobartt / safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
flax jax serialization safetensors objax haiku flax-serialization jax-serialization objax-serialization haiku-serialization safejax huggingface
Language:Python 38
mamaj / cnn-featurevis-ece421
Programming assignment for UofT ECE421, Fall 2020: CNN Feature Visualization using jax and objax.
jax objax colab-notebook vgg19 feature-visualization
Language:Jupyter Notebook 1

rwightman / efficientnet-jax