flax-serialization

There are 0 repository under flax-serialization topic.

alvarobartt / safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
flax jax serialization safetensors objax haiku flax-serialization jax-serialization objax-serialization haiku-serialization safejax huggingface
Language:Python 47