mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/notebooks
History
Yash Katariya 160bbe12d3 Fix shard_map docs build 
PiperOrigin-RevId: 746033054
2025-04-10 08:04:52 -07:00
..
autodiff_cookbook.ipynb
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
autodiff_cookbook.md
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
autodiff_remat.ipynb
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
autodiff_remat.md
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
Common_Gotchas_in_JAX.ipynb
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
Common_Gotchas_in_JAX.md
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
convolutions.ipynb
Fix convolution example (kernel should be OIHW, not IOHW).
2025-03-03 09:32:22 -08:00
convolutions.md
Fix convolution example (kernel should be OIHW, not IOHW).
2025-03-03 09:32:22 -08:00
Custom_derivative_rules_for_Python_code.ipynb
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
Custom_derivative_rules_for_Python_code.md
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
Distributed_arrays_and_automatic_parallelization.ipynb
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
Distributed_arrays_and_automatic_parallelization.md
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
explicit-sharding.ipynb
mixing modes
2025-03-14 18:23:27 -07:00
explicit-sharding.md
mixing modes
2025-03-14 18:23:27 -07:00
Neural_Network_and_Data_Loading.ipynb
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
Neural_Network_and_Data_Loading.md
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
neural_network_with_tfds_data.ipynb
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
neural_network_with_tfds_data.md
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
README.md
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
shard_map.ipynb
Fix shard_map docs build
2025-04-10 08:04:52 -07:00
shard_map.md
Fix shard_map docs build
2025-04-10 08:04:52 -07:00
thinking_in_jax.ipynb
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
thinking_in_jax.md
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
vmapped_log_probs.ipynb
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
vmapped_log_probs.md
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
Writing_custom_interpreters_in_Jax.ipynb
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
Writing_custom_interpreters_in_Jax.md
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00

README.md

For instructions on how to change and test notebooks, see Update Documentation.