Yash Katariya ca2d1584f8 Remove mesh_utils.create_device_mesh from docs ...

PiperOrigin-RevId: 687695419

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

..

263-prng.md

DOC: read-through and edit the new jax tutorials

2024-03-20 18:18:08 -07:00

2026-custom-derivatives.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

4008-custom-vjp-update.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

4410-omnistaging.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

9263-typed-keys.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

9407-type-promotion.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

9407-type-promotion.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

9419-jax-versioning.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

10657-sequencing-effects.md

Fixed Typos in JEP doc files

2024-03-12 09:58:13 +05:30

11830-new-remat-checkpoint.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

12049-type-annotations.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

14273-shard-map.md

Remove mesh_utils.create_device_mesh from docs

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

15856-jex.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

17111-shmap-transpose.md

Fix typos across the package

2023-09-22 14:54:31 -07:00

18137-numpy-scipy-scope.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

index.rst

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