mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/notebooks
History
8bitmp3 7e2e4a08e4 Add jit/shard_map banner to xmap docs 
Add jit/shard_map banner to xmap docs
2023-10-12 22:06:15 +00:00
..
autodiff_cookbook.ipynb
Replace references to DeviceArray with Array.
2023-08-18 17:46:00 -04:00
autodiff_cookbook.md
CI: update jupytext to v0.14.7
2023-07-24 11:51:45 -07:00
autodiff_remat.ipynb
Fix typos across the package
2023-09-22 14:54:31 -07:00
autodiff_remat.md
Fix typos across the package
2023-09-22 14:54:31 -07:00
Common_Gotchas_in_JAX.ipynb
Replace references to DeviceArray with Array.
2023-08-18 17:46:00 -04:00
Common_Gotchas_in_JAX.md
Replace references to DeviceArray with Array.
2023-08-18 17:46:00 -04:00
convolutions.ipynb
Fix typos across the package
2023-09-22 14:54:31 -07:00
convolutions.md
Fix typos across the package
2023-09-22 14:54:31 -07:00
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
Replace references to DeviceArray with Array.
2023-08-18 17:46:00 -04:00
Distributed_arrays_and_automatic_parallelization.ipynb
Add Kaggle TPU VM link, update Distributed Arrays and Auto Parallelization guide
2023-08-21 21:17:02 +00:00
Distributed_arrays_and_automatic_parallelization.md
Add Kaggle TPU VM link, update Distributed Arrays and Auto Parallelization guide
2023-08-21 21:17:02 +00:00
external_callbacks.ipynb
Fix typos across the package
2023-09-22 14:54:31 -07:00
external_callbacks.md
Fix typos across the package
2023-09-22 14:54:31 -07:00
How_JAX_primitives_work.ipynb
Remove uses of XLA translation rules.
2023-08-15 12:53:36 -07:00
How_JAX_primitives_work.md
Remove uses of XLA translation rules.
2023-08-15 12:53:36 -07:00
Neural_Network_and_Data_Loading.ipynb
Simplify the torch data loader collate function using tree_map.
2023-10-04 14:59:06 -04:00
Neural_Network_and_Data_Loading.md
Simplify the torch data loader collate function using tree_map.
2023-10-04 14:59:06 -04:00
neural_network_with_tfds_data.ipynb
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
neural_network_with_tfds_data.md
CI: update jupytext to v0.14.7
2023-07-24 11:51:45 -07:00
quickstart.ipynb
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
quickstart.md
CI: update jupytext to v0.14.7
2023-07-24 11:51:45 -07:00
README.md
Added more documentation for how to fix notebook build failures (#2404)
2020-03-12 10:59:30 +01:00
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
DOC: re-enable execution of thinking_in_jax.ipynb
2023-08-24 09:23:26 -07:00
vmapped_log_probs.ipynb
Replace references to DeviceArray with Array.
2023-08-18 17:46:00 -04:00
vmapped_log_probs.md
CI: update jupytext to v0.14.7
2023-07-24 11:51:45 -07:00
Writing_custom_interpreters_in_Jax.ipynb
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
Writing_custom_interpreters_in_Jax.md
CI: update jupytext to v0.14.7
2023-07-24 11:51:45 -07:00
xmap_tutorial.ipynb
Add jit/shard_map banner to xmap docs
2023-10-12 22:06:15 +00:00
xmap_tutorial.md
Add jit/shard_map banner to xmap docs
2023-10-12 22:06:15 +00:00

README.md

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