mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add doc for jax.experimental.serialize_executable module

Browse Source
This commit is contained in:
Frederic Bastien 2024-05-14 13:36:35 -07:00
parent 9af0a317a5
commit 936ec98380
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