carlosgmartin
|
bc43b00d8f
|
Add navigation breadcrumbs to docs.
|
2025-03-12 16:52:34 -04:00 |
|
Justin Fu
|
2b2d7cda98
|
[Pallas] Update TPU documentation
|
2024-12-06 10:59:11 -08:00 |
|
barnesjoseph
|
bf7f9aa8f2
|
Adds Google Sans font
|
2024-11-20 16:44:41 -08:00 |
|
barnesjoseph
|
d4316b5760
|
Adds font fallbacks
|
2024-11-18 14:46:10 -08:00 |
|
barnesjoseph
|
81cdc882ae
|
DOC: update main landing page style
Co-authored-by: Jake VanderPlas <jakevdp@google.com>
|
2024-11-15 13:44:31 -08:00 |
|
Jake VanderPlas
|
72a4692b94
|
doc: link directly to installation on the main page
|
2024-11-13 12:44:05 -08:00 |
|
Justin Fu
|
4bce4f6452
|
[Pallas] Add block-sparse kernel tutorial
|
2024-09-19 12:23:03 -07:00 |
|
Justin Fu
|
51a666fb8c
|
[Pallas] Update Pallas docs with new figures and TPUCompilerParams
|
2024-09-06 14:30:29 -07:00 |
|
Jake VanderPlas
|
f04a35bda5
|
Update jax landing page
|
2024-08-20 13:03:08 -07:00 |
|
Justin Fu
|
bdb03309b5
|
Merge branch 'main' into pallas_distr_docs
|
2024-08-08 17:36:36 -07:00 |
|
Justin Fu
|
dcd186f552
|
[Pallas] Add pallas distributed computation tutorial
|
2024-08-08 17:34:35 -07:00 |
|
George Necula
|
b58ff2ba20
|
[shape_poly] Add documentation for shape polymorphism
This involved writing some new content and also moving and adapting
the documentation that existed as part of the jax2tf
README file:
https://github.com/google/jax/blob/main/jax/experimental/jax2tf/README.md#shape-polymorphic-conversion
|
2024-06-15 18:20:54 +03:00 |
|
8bitmp3
|
15a3debee8
|
Add Distributed data loading doc
|
2024-05-16 23:12:13 +00:00 |
|
Jake VanderPlas
|
e21286d6f0
|
JEP 9263: Typed keys & pluggable RNGs
Co-authored-by: Roy Frostig <frostig@google.com>
|
2023-09-20 11:26:13 -07:00 |
|
Sharad Vikram
|
96e2d93f53
|
[Pallas] Add Pallas design doc
|
2023-08-03 16:02:07 -07:00 |
|
Sharad Vikram
|
7f1ef32ba3
|
Add initial documentation for Pallas
|
2023-08-03 12:30:19 -07:00 |
|
Jake VanderPlas
|
55d9c06267
|
DOC: update sphinx & sphinx-autodoc-typehints
|
2023-03-01 11:03:20 -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 |
|
Jake VanderPlas
|
358363e17f
|
JEP 12049: Type Annotation Roadmap
|
2022-09-13 09:14:48 -07:00 |
|
Sharad Vikram
|
4386a0f909
|
Add debugging tools under jax.debug and documentation
Co-authored-by: Matthew Johnson <mattjj@google.com>
Co-authored-by: Lena Martens <lenamartens@google.com>
|
2022-07-28 20:07:26 -07:00 |
|
Sharad Vikram
|
76669835ba
|
Add an option to create a perfetto link in the JAX profiler
|
2022-06-01 15:48:29 -07:00 |
|
Jake VanderPlas
|
e82f232ea9
|
Fix nomenclature in type promotion doc
|
2022-02-15 13:26:06 -08:00 |
|
Tina Jia
|
add8300372
|
added images for pjit tutorial
|
2021-09-08 23:29:20 +00:00 |
|
jax authors
|
860d1b3de4
|
Merge pull request #4815 from j-towns:docs-favicon
PiperOrigin-RevId: 341325726
|
2020-11-08 19:06:24 -08:00 |
|
Jamie Townsend
|
46ca7a00ce
|
Rm gridlines from favicon (thanks @shoyer)
|
2020-11-08 19:12:40 +00:00 |
|
Jake VanderPlas
|
86f4535393
|
DOC: add lattice to type promotion doc
|
2020-11-06 09:13:28 -08:00 |
|
Jamie Townsend
|
62683959b1
|
[DOCS] Add JAX favicon to sphinx
|
2020-11-06 13:43:18 +00:00 |
|
Peter Hawkins
|
5116fd47aa
|
Add a heap profiler API and document it. (#3576)
|
2020-06-26 17:09:09 -04:00 |
|
Skye Wanderman-Milne
|
8fe6da08d4
|
Add instructions for how to use the TensorBoard profiler to the profiling docs. (#3481)
|
2020-06-22 17:33:23 -07:00 |
|
Stephan Hoyer
|
515e07bdf2
|
White background
|
2019-08-05 13:08:17 -07:00 |
|
Stephan Hoyer
|
8c8222a319
|
Add the JAX logo to sphinx docs
|
2019-08-05 12:34:29 -07:00 |
|