flax-serialization topic

List flax-serialization repositories

safejax

38
Stars
2
Forks
Watchers

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

verified publisher icon alvarobartt