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
Peter Hawkins 816ba91263 Use lower-case PEP 585 names for types. 
Issue https://github.com/google/jax/issues/16537

PiperOrigin-RevId: 542969282
2023-06-23 15:12:14 -07:00
..
autodiff_cookbook.ipynb
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
autodiff_cookbook.md
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
autodiff_remat.ipynb
Use lower-case PEP 585 names for types.
2023-06-23 15:12:14 -07:00
autodiff_remat.md
Use lower-case PEP 585 names for types.
2023-06-23 15:12:14 -07:00
Common_Gotchas_in_JAX.ipynb
Change to simpler import for jax.config
2023-04-21 11:51:22 -07:00
Common_Gotchas_in_JAX.md
Change to simpler import for jax.config
2023-04-21 11:51:22 -07:00
convolutions.ipynb
Fix typo: "__W__ - spatial height" -> "__W__ - spatial width"
2023-04-11 04:37:26 -07:00
convolutions.md
Fix typo: "__W__ - spatial height" -> "__W__ - spatial width"
2023-04-11 04:37:26 -07:00
Custom_derivative_rules_for_Python_code.ipynb
fix custom_jvp docs typo
2023-04-15 14:51:33 -07:00
Custom_derivative_rules_for_Python_code.md
fix custom_jvp docs typo
2023-04-15 14:51:33 -07: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
CI: update jupytext version
2023-01-23 14:42:03 -08:00
external_callbacks.ipynb
Add a CI presubmit that renders the documentation.
2023-06-20 09:29:25 -04:00
external_callbacks.md
Add documentation for callback functions
2023-04-07 07:05:46 -07:00
How_JAX_primitives_work.ipynb
Mark jax.abstract_arrays as deprecated
2023-06-07 23:36:40 -07:00
How_JAX_primitives_work.md
Mark jax.abstract_arrays as deprecated
2023-06-07 23:36:40 -07:00
Neural_Network_and_Data_Loading.ipynb
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
Neural_Network_and_Data_Loading.md
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05: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
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
quickstart.ipynb
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
quickstart.md
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05: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
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
thinking_in_jax.md
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
vmapped_log_probs.ipynb
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
vmapped_log_probs.md
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05: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
Add "Open in Kaggle" buttons to Jupyter notebooks.
2023-03-01 13:15:42 -05:00
xmap_tutorial.ipynb
Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.
2023-02-15 21:03:03 -08:00
xmap_tutorial.md
Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.
2023-02-15 21:03:03 -08:00

README.md

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