18 Commits

Author SHA1 Message Date
8bitmp3
0cf040c9a1 Add/update JAX Advanced Tutorials docs, ToC structure 2024-09-20 23:06:54 +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
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
Dan Foreman-Mackey
0b4800a193 Add ffi_call tutorial
Building on #21925, this tutorial demonstrates the use of the FFI using
`ffi_call` with a simple example. I don't think this should cover all of
the most advanced use cases, but it should be sufficient for the most
common examples. I think it would be useful to eventually replace the
existing CUDA tutorial, but I'm not sure that it'll get there in the
first draft.

As an added benefit, this also runs a simple test (akin to
`docs/cuda_custom_call`) which actually executes using a tool chain that
open source users would use in practice.
2024-08-01 15:36:32 -04:00
George Necula
105cc9a103 [export] Add documentation for jax.export 2024-06-12 19:44:47 +02:00
Jake VanderPlas
10ed827fe9 DOC: replace old tutorials with new content 2024-05-03 12:20:06 -07:00
jax authors
afb775c168 Jax persistent compilation cache user guide.
This user guide covers using the cache on local filesystems
and Google Cloud.

PiperOrigin-RevId: 623236335
2024-04-09 11:48:33 -07:00
Sharad Vikram
2f8d5cebff [Pallas TPU] Add Pipelining and BlockSpecs documentation for Pallas TPU 2024-02-28 19:17:58 -08:00
Matthew Johnson
664e834784 draft docs on gpu performance tuning
Co-authored-by: Tao Wang <wangtao@google.com>
2023-10-30 20:33:56 -07:00
Sharad Vikram
7f1ef32ba3 Add initial documentation for Pallas 2023-08-03 12:30:19 -07:00
Jake VanderPlas
ec63d699e9 DOC: fix headings in user_guides 2023-03-30 14:39:25 -07:00
Ravin Kumar
8c2549519b
Update user_guides.rst
Fix minor typo
2023-03-26 17:21:35 -07:00
Ravin Kumar
18b251e251 Add notes and headeers to user guides 2023-02-10 14:17:15 -08:00
Jake VanderPlas
01b5799239 DOC: re-organize notes section 2023-01-30 11:57:06 -08:00
Jake VanderPlas
fd71794633 DOC: organize content in advanced guide 2023-01-26 08:20:53 -08:00
8bitmp3
d6cc2bdb22 Make JAX Debugging and Profiling guides more visible, move Profiling to User Guides from Notes 2023-01-19 23:05:24 +00: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