mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 01:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/notebooks
History
Jake VanderPlas 10ed827fe9 DOC: replace old tutorials with new content
2024-05-03 12:20:06 -07:00
..
autodiff_cookbook.ipynb
update notebooks to new-style typed PRNG keys
2024-03-07 12:40:10 -08:00
autodiff_cookbook.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
autodiff_remat.ipynb
Fix typos across the package
2023-09-22 14:54:31 -07:00
autodiff_remat.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
Common_Gotchas_in_JAX.ipynb
Avoid 'from jax import config' imports
2024-04-11 13:23:27 -07:00
Common_Gotchas_in_JAX.md
Avoid 'from jax import config' imports
2024-04-11 13:23:27 -07:00
convolutions.ipynb
update notebooks to new-style typed PRNG keys
2024-03-07 12:40:10 -08:00
convolutions.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
Custom_derivative_rules_for_Python_code.ipynb
Replace references to DeviceArray with Array.
2023-08-18 17:46:00 -04:00
Custom_derivative_rules_for_Python_code.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
Distributed_arrays_and_automatic_parallelization.ipynb
Update JAX official doc: point out that the device numbers are not in numerical order because of the underlying torus hardware topology.
2024-04-16 11:37:19 -07:00
Distributed_arrays_and_automatic_parallelization.md
Update JAX official doc: point out that the device numbers are not in numerical order because of the underlying torus hardware topology.
2024-04-16 11:37:19 -07:00
external_callbacks.ipynb
Fix typos across the package
2023-09-22 14:54:31 -07:00
external_callbacks.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
How_JAX_primitives_work.ipynb
fixed typo 'primtive' as 'primitive' in How_JAX_primitives_work.ipynb
2023-10-13 21:05:43 +05:45
How_JAX_primitives_work.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
Neural_Network_and_Data_Loading.ipynb
DOC: replace old tutorials with new content
2024-05-03 12:20:06 -07:00
Neural_Network_and_Data_Loading.md
DOC: replace old tutorials with new content
2024-05-03 12:20:06 -07:00
neural_network_with_tfds_data.ipynb
DOC: replace old tutorials with new content
2024-05-03 12:20:06 -07:00
neural_network_with_tfds_data.md
DOC: replace old tutorials with new content
2024-05-03 12:20:06 -07:00
README.md
Added more documentation for how to fix notebook build failures (#2404)
2020-03-12 10:59:30 +01:00
shard_map.ipynb
fix typos in shard_map tutorial
2024-03-12 11:58:26 -07:00
shard_map.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
thinking_in_jax.ipynb
DOC: re-enable execution of thinking_in_jax.ipynb
2023-08-24 09:23:26 -07:00
thinking_in_jax.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
vmapped_log_probs.ipynb
update notebooks to new-style typed PRNG keys
2024-03-07 12:40:10 -08:00
vmapped_log_probs.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
Writing_custom_interpreters_in_Jax.ipynb
update notebooks to new-style typed PRNG keys
2024-03-07 12:40:10 -08:00
Writing_custom_interpreters_in_Jax.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30
xmap_tutorial.ipynb
Add a deprecation warning to the xmap function and the corresponding tutorial
2024-03-11 21:31:32 +00:00
xmap_tutorial.md
Updated all the pre-commit hooks versions.
2024-04-08 00:59:02 +05:30

README.md

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