alvarobartt / safejax

Serialize JAX, Flax, Haiku, or Objax model params with πŸ€—`safetensors`
β˜†42Updated 5 months ago

Related projects β“˜

Alternatives and complementary repositories for safejax