mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-23 21:56:04 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/notebooks
History
8bitmp3 2bf6d3dace Update jax.xmap notebook title (Named axes and easy-to-revise parallelism with xmap)
2023-01-07 00:05:46 +00:00
..
autodiff_cookbook.ipynb
Fix book links
2022-10-03 13:59:10 +02:00
autodiff_cookbook.md
Fix book links
2022-10-03 13:59:10 +02:00
Common_Gotchas_in_JAX.ipynb
Create JAX Enhancement Proposals (JEPs).
2022-08-08 16:13:58 -04:00
Common_Gotchas_in_JAX.md
pre-commit hook: update flake8, mypy, & jupytext
2022-08-15 15:32:45 -07:00
convolutions.ipynb
docs: avoid deprecated matplotlib axis creation
2022-09-19 12:55:18 -07:00
convolutions.md
docs: avoid deprecated matplotlib axis creation
2022-09-19 12:55:18 -07:00
Custom_derivative_rules_for_Python_code.ipynb
remove cotangent negation in custom VJP example
2022-11-22 17:55:22 -08:00
Custom_derivative_rules_for_Python_code.md
remove cotangent negation in custom VJP example
2022-11-22 17:55:22 -08:00
Distributed_arrays_and_automatic_parallelization.ipynb
Remove experimental endpoints and update to point to 0.4.1
2022-12-13 11:53:56 -08:00
Distributed_arrays_and_automatic_parallelization.md
Remove experimental endpoints and update to point to 0.4.1
2022-12-13 11:53:56 -08:00
external_callbacks.ipynb
DOC: add example of pure_callback with custom_jvp
2022-12-09 12:43:04 -08:00
external_callbacks.md
DOC: add example of pure_callback with custom_jvp
2022-12-09 12:43:04 -08:00
How_JAX_primitives_work.ipynb
Port remaining translation rules inside JAX to new style.
2021-10-19 09:48:37 -07:00
How_JAX_primitives_work.md
pre-commit hook: update flake8, mypy, & jupytext
2022-08-15 15:32:45 -07:00
Neural_Network_and_Data_Loading.ipynb
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
Neural_Network_and_Data_Loading.md
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
neural_network_with_tfds_data.ipynb
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
neural_network_with_tfds_data.md
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
quickstart.ipynb
DOC: pre-execute the quickstart notebook on GPU
2022-12-06 13:24:02 -08:00
quickstart.md
DOC: pre-execute the quickstart notebook on GPU
2022-12-06 13:24:02 -08: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: fix output repr in thinking_in_jax
2022-06-21 12:54:50 -07:00
thinking_in_jax.md
pre-commit hook: update flake8, mypy, & jupytext
2022-08-15 15:32:45 -07:00
vmapped_log_probs.ipynb
Rename master to main in embedded links.
2021-06-18 10:00:01 +03:00
vmapped_log_probs.md
pre-commit hook: update flake8, mypy, & jupytext
2022-08-15 15:32:45 -07:00
Writing_custom_interpreters_in_Jax.ipynb
Update custom interpreter tutorial
2022-05-11 11:11:47 -07:00
Writing_custom_interpreters_in_Jax.md
pre-commit hook: update flake8, mypy, & jupytext
2022-08-15 15:32:45 -07:00
xmap_tutorial.ipynb
Update jax.xmap notebook title (Named axes and easy-to-revise parallelism with xmap)
2023-01-07 00:05:46 +00:00
xmap_tutorial.md
Update jax.xmap notebook title (Named axes and easy-to-revise parallelism with xmap)
2023-01-07 00:05:46 +00:00

README.md

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