alvarobartt / safejax

Serialize JAX, Flax, Haiku, or Objax model params with ๐Ÿค—`safetensors`
โ˜†42Updated 5 months ago

Related projects โ“˜

Alternatives and complementary repositories for safejax