rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-22 20:56:05 +00:00

History

Yash Katariya ca2d1584f8 Remove mesh_utils.create_device_mesh from docs

PiperOrigin-RevId: 687695419

2024-10-19 15:48:42 -07:00

autodiff_cookbook.ipynb

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

autodiff_cookbook.md

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

autodiff_remat.ipynb

docs: prefer "summary" to "tl;dr"

2024-08-28 11:04:14 -07:00

autodiff_remat.md

docs: prefer "summary" to "tl;dr"

2024-08-28 11:04:14 -07:00

Common_Gotchas_in_JAX.ipynb

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

Common_Gotchas_in_JAX.md

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

convolutions.ipynb

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

convolutions.md

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

Custom_derivative_rules_for_Python_code.ipynb

CI: enable additional ruff formatting checks

2024-10-16 16:09:54 -07:00

Custom_derivative_rules_for_Python_code.md

CI: enable additional ruff formatting checks

2024-10-16 16:09:54 -07:00

Distributed_arrays_and_automatic_parallelization.ipynb

Remove mesh_utils.create_device_mesh from docs

2024-10-19 15:48:42 -07:00

Distributed_arrays_and_automatic_parallelization.md

Remove mesh_utils.create_device_mesh from docs

2024-10-19 15:48:42 -07:00

Neural_Network_and_Data_Loading.ipynb

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

Neural_Network_and_Data_Loading.md

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

neural_network_with_tfds_data.ipynb

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

neural_network_with_tfds_data.md

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -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

Remove mesh_utils.create_device_mesh from docs

2024-10-19 15:48:42 -07:00

shard_map.md

Remove mesh_utils.create_device_mesh from docs

2024-10-19 15:48:42 -07:00

thinking_in_jax.ipynb

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

thinking_in_jax.md

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

vmapped_log_probs.ipynb

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

vmapped_log_probs.md

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

Writing_custom_interpreters_in_Jax.ipynb

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

Writing_custom_interpreters_in_Jax.md

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

README.md

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