Orbax Export
conda install anaconda::orbax-export
Orbax includes a serialization library for JAX users, enabling the exporting of JAX models to the TensorFlow SavedModel format.