Dougal
e8f43d1cef
Explicit sharding docs
2025-03-14 16:33:30 -04:00
Yury Kirpichev
f9204e6311
Add md page about XLA&XLA_FLAGS in JAX
2024-09-05 09:27:32 +00:00
Roy Frostig
12eebfe8d4
docs: reorganize sections
...
* Create "extension guides" section
* Sort developer notes into subsections
* Move examples from advanced section into user guides
* Reorder some listings, adjust some titles
2024-08-13 21:33:45 -07:00
Roy Frostig
3c223cd253
docs: tidy up titles and headings
...
This shortens some titles and makes them more consistent. It also
removes "JAX" from several titles ("in JAX", "for JAX", "JAX's",
etc.). Since these are JAX docs, that ought to be clear from context.
2024-08-13 11:53:57 -07:00
Roy Frostig
2644299f7e
docs: sentence case index and sub-index headings
...
We currently use both forms, so for consistency (and easier reading),
pick this one.
2024-08-12 13:52:43 -07:00
Yash Katariya
0d5dae09ff
Delete xmap and the jax.experimental.maps module. It's been 5 months since its deprecation (more than the standard 3 months deprecation period).
...
PiperOrigin-RevId: 655614395
2024-07-24 10:24:09 -07:00
8bitmp3
15a3debee8
Add Distributed data loading doc
2024-05-16 23:12:13 +00:00
Matthew Johnson
8b219d5f7b
[shard-map] add user tutorial
2024-01-18 15:30:13 -08:00
Matthew Johnson
c22da81d5d
fixes from reviewers
2023-02-23 15:06:55 -08:00
Jake VanderPlas
fd71794633
DOC: organize content in advanced guide
2023-01-26 08:20:53 -08:00
Jake VanderPlas
e5f0103895
Reorganization among notes, user guides, advanced guides
2023-01-19 09:52:43 -08:00
Ravin Kumar
2257e2075d
Update doc landing page
...
Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com>
Co-authored-by: Jake VanderPlas <jakevdp@google.com>
2023-01-13 12:45:28 -08:00
