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

Add jax.make_array_from_process_local_data to the docs. Preview: https://jax--21444.org.readthedocs.build/en/21444/_autosummary/jax.make_array_from_process_local_data.html#jax.make_array_from_process_local_data

Browse Source 
PiperOrigin-RevId: 637660828
This commit is contained in:
Yash Katariya 2024-05-27 09:40:49 -07:00 committed by jax authors
parent 24509f325c
commit 08b1cef00a
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/jax.rst
View File

@ -113,6 +113,7 @@ jax.Array (:code:`jax.Array`)
Array
make_array_from_callback
make_array_from_single_device_arrays
make_array_from_process_local_data
Vectorization (:code:`vmap`)
----------------------------