safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗safetensors
41 stars
2 watching
5 forks
Language: Python
last commit: 4 months ago
Linked from 1 awesome list
flaxflax-serializationhaikuhaiku-serializationhuggingfacejaxjax-serializationobjaxobjax-serializationsafejaxsafetensorsserialization
n2cholas/awesome-jax