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