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