https://github.com/alvarobartt/safejax
Python42
5 months ago
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
MIT License