#

safejax

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 safejax 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 safejax topic, visit your repo's landing page and select "manage topics."