safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
Python28mit
8 months ago
flaxflax-serializationhaiku
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
Python28mit
8 months ago
flaxflax-serializationhaiku