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