safejax

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

Creator

alvarobartt

Related apps