Michael Hudgins
2e808f2836
Merge pull request #26279 from MichaelHudgins:tsan-resultstore
...
PiperOrigin-RevId: 723918760
2025-02-06 14:55:57 +00:00
Yash Katariya
83ca7c33d2
Don't create a mesh inside jit in sharded-computation tutorial
...
PiperOrigin-RevId: 707288222
2024-12-17 16:11:00 -08:00
jax authors
dfa4e2413c
Merge pull request #23643 from TomAugspurger:fix/doc-typos
...
PiperOrigin-RevId: 675133675
2024-09-16 07:07:26 -07:00
enerrio
b8d135aa05
fix small typos in docs
2024-09-14 13:53:19 -07:00
Tom Augspurger
fcc8c3759d
Fixed func ref in shared-computation
2024-09-14 15:22:12 -05:00
Jake VanderPlas
7569dd5438
Update sharded-computation doc to use make_mesh()
2024-09-03 16:04:23 -07:00
Roy Frostig
7b29791200
docs: shorten/clarify some page titles
...
Shorten the titles for derivative rules and manual parallelism, and go
for canonical wording in the parallel programming intro (typically we
"shard" data, and "partition" computation, as part of parallel
programming).
2024-08-28 11:22:06 -07:00
Jake VanderPlas
09fd345de9
pre-commit: update hooks & pin using hashes
2024-08-27 15:23:13 -07:00
Jake VanderPlas
68be5b5085
CI: update ruff to v0.6.1
2024-08-27 14:54:11 -07:00
jax authors
fc1e1d4a65
Add freshness metablock to JAX OSS docs.
...
PiperOrigin-RevId: 645508135
2024-06-21 14:50:49 -07:00
8bitmp3
9ea3fcbab3
Upgrade JAX Parallelism Sharded Computation 101 doc
2024-05-10 21:24:16 +00:00
Jake VanderPlas
10ed827fe9
DOC: replace old tutorials with new content
2024-05-03 12:20:06 -07:00
