flax-serialization topic
List
flax-serialization repositories
safejax
41
Stars
5
Forks
Watchers
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
