jax-serialization topic
List
jax-serialization repositories
safejax
38
Stars
2
Forks
Watchers
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`