flax-serialization
There are 1 repositories under flax-serialization topic.
alvarobartt/safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
There are 1 repositories under flax-serialization topic.
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`