14 Commits

Author SHA1 Message Date
George Necula
bfdf8f4bd3 [pallas] Added more documentation for grid and BlockSpec.
The starting point was the text in pipelining.md, where I
replaced it now with a reference to the separate grid and BlockSpec
documentation.

The grids and BlockSpecs are also documented in the quickstart.md,
which I mostly left alone because it was good enough for a
simple example.

I have also attempted to add a few docstrings.
2024-06-29 14:43:48 +03:00
George Necula
8528f5127d [pallas] Break long lines in the Pallas docs
No content changes.
2024-06-25 13:30:17 +03:00
jax authors
fc1e1d4a65 Add freshness metablock to JAX OSS docs.
PiperOrigin-RevId: 645508135
2024-06-21 14:50:49 -07:00
rajasekharporeddy
61c64c10f8 Fixed Several Typos
Fixed Typos in JEP doc files

Revert "Fixed Typos in JEP doc files"

This reverts commit c2a16950e0fc1b32971168501d183991e2394b5d.

revert two changes

reverted one change in advanced-autodiff

revert one change in parallelism

sync notebooks
2024-03-12 00:37:46 +05:30
Roy Frostig
fe3e798f82 update pallas quickstart to new-style typed PRNG keys 2024-03-07 12:40:10 -08:00
Matthew Johnson
efe78c53ec improve block matrix alignment in pallas docs 2024-01-10 22:01:16 -08:00
Jeremy Kun
2e6e5da49b docs/pallas: remove list from out_specs 2024-01-09 15:47:18 -08:00
Jeremy Kun
4ecbed1322 docs/pallas: sync jupyter notebook 2024-01-09 15:00:42 -08:00
Alexey Radul
1f184541e4 Ah, jupytext converts from markdown to a notebook, not vice versa. 2023-11-09 11:11:21 -05:00
Alexey Radul
785d9fa14c Lightly proofread Pallas documentation. 2023-11-09 10:57:47 -05:00
Jake VanderPlas
4a5bd9e046 Fix typos across the package 2023-09-22 14:54:31 -07:00
Sharad Vikram
00bc7cf6f7 [Pallas] Fix out_specs in Pallas quickstart 2023-09-21 17:58:57 -07:00
Sharad Vikram
28af1861ee [Pallas] Fix rendering of math in quickstart 2023-08-05 00:20:00 -07:00
Sharad Vikram
7f1ef32ba3 Add initial documentation for Pallas 2023-08-03 12:30:19 -07:00