#

jax-serialization

Here is 1 public repository matching this topic...

alvarobartt / safejax

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

serialization flax haiku jax huggingface objax safetensors jax-serialization flax-serialization objax-serialization haiku-serialization safejax

Updated May 31, 2024
Python

Improve this page

Add a description, image, and links to the jax-serialization topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the jax-serialization topic, visit your repo's landing page and select "manage topics."