mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-25 00:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/notebooks
History
Dan Foreman-Mackey 705eed3388 Fixing dtype canonicalization in sharp edges tutorial. 
As reported in https://github.com/google/jax/issues/22493, the sharp
edges tutorial doesn't seem to actually enable x64 when it says it does.

Fixes https://github.com/google/jax/issues/22493
2024-07-22 15:02:02 -04:00
..
autodiff_cookbook.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
autodiff_cookbook.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
autodiff_remat.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
autodiff_remat.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
Common_Gotchas_in_JAX.ipynb
Fixing dtype canonicalization in sharp edges tutorial.
2024-07-22 15:02:02 -04:00
Common_Gotchas_in_JAX.md
Fixing dtype canonicalization in sharp edges tutorial.
2024-07-22 15:02:02 -04:00
convolutions.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
convolutions.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
Custom_derivative_rules_for_Python_code.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
Custom_derivative_rules_for_Python_code.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
Distributed_arrays_and_automatic_parallelization.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
Distributed_arrays_and_automatic_parallelization.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
external_callbacks.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
external_callbacks.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
How_JAX_primitives_work.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
How_JAX_primitives_work.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
Neural_Network_and_Data_Loading.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
Neural_Network_and_Data_Loading.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
neural_network_with_tfds_data.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
neural_network_with_tfds_data.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -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
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
shard_map.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
thinking_in_jax.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
thinking_in_jax.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
vmapped_log_probs.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
vmapped_log_probs.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
Writing_custom_interpreters_in_Jax.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
Writing_custom_interpreters_in_Jax.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
xmap_tutorial.ipynb
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00
xmap_tutorial.md
Add freshness metablock to JAX OSS docs.
2024-06-21 14:50:49 -07:00

README.md

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