mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #21231 from nouiz:doc_experimental_serialize_executable

Browse Source 
PiperOrigin-RevId: 635428472
This commit is contained in:
jax authors 2024-05-20 05:39:21 -07:00
commit bb616eff8a
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/jax.experimental.rst
View File

@ -26,6 +26,7 @@ Experimental Modules
jax.experimental.compilation_cache
jax.experimental.key_reuse
jax.experimental.mesh_utils
jax.experimental.serialize_executable
jax.experimental.shard_map
Experimental APIs

13
docs/jax.experimental.serialize_executable.rst Normal file
View File

@ -0,0 +1,13 @@
``jax.experimental.serialize_executable`` module
================================================
.. automodule:: jax.experimental.serialize_executable
API
---
.. autosummary::
:toctree: _autosummary
serialize
deserialize_and_load