https://github.com/alvarobartt/safejax

Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`

flax flax-serialization haiku haiku-serialization huggingface jax jax-serialization objax objax-serialization safejax safetensors serialization

Last synced: 11 Apr 2025