jax-serialization topic

List jax-serialization repositories

safejax

41

Stars

5

Forks

Watchers

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

flax-serialization

haiku-serialization