mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 07:26:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/notebooks
History
Roy Frostig fcce6b102c remove cotangent negation in custom VJP example 
This was originally intended to show that we can change the VJP by
customizing it, but the algebraic incorrectness is confusing.
2022-11-22 17:55:22 -08: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
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
jax_Array.ipynb
jax.Array tutorial
2022-11-15 16:49:17 -08:00
jax_Array.md
jax.Array tutorial
2022-11-15 16:49:17 -08: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
quickstart: fix outdated TPU reference
2022-03-21 09:41:10 -07:00
quickstart.md
pre-commit hook: update flake8, mypy, & jupytext
2022-08-15 15:32: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: 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
Remove a TODO from the xmap tutorial
2022-10-26 15:44:06 +00:00
xmap_tutorial.md
Remove a TODO from the xmap tutorial
2022-10-26 15:44:06 +00:00

README.md

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