Justin Fu 4bce4f6452 [Pallas] Add block-sparse kernel tutorial

2024-09-19 12:23:03 -07:00

..

distributed_data_loading

Add Distributed data loading doc

2024-05-16 23:12:13 +00:00

pallas

[Pallas] Add block-sparse kernel tutorial

2024-09-19 12:23:03 -07:00

debugger.gif

Add debugging tools under jax.debug and documentation

2022-07-28 20:07:26 -07:00

device_memory_profile_leak1.svg

Add a heap profiler API and document it. (#3576)

2020-06-26 17:09:09 -04:00

device_memory_profile_leak2.svg

Add a heap profiler API and document it. (#3576)

2020-06-26 17:09:09 -04:00

device_memory_profile.svg

Add a heap profiler API and document it. (#3576)

2020-06-26 17:09:09 -04:00

favicon.png

Rm gridlines from favicon (thanks @shoyer)

2020-11-08 19:12:40 +00:00

jax_logo_250px.png

Add the JAX logo to sphinx docs

2019-08-05 12:34:29 -07:00

mesh.jpg

added images for pjit tutorial

2021-09-08 23:29:20 +00:00

multi_host.jpg

added images for pjit tutorial

2021-09-08 23:29:20 +00:00

partition_spec_none_y.png

added images for pjit tutorial

2021-09-08 23:29:20 +00:00

partition_spec_x_none.png

added images for pjit tutorial

2021-09-08 23:29:20 +00:00

partition_spec_x_y.png

added images for pjit tutorial

2021-09-08 23:29:20 +00:00

partition_spec_xy.png

added images for pjit tutorial

2021-09-08 23:29:20 +00:00

partition_spec_y_none.png

added images for pjit tutorial

2021-09-08 23:29:20 +00:00

perfetto.png

Add an option to create a perfetto link in the JAX profiler

2022-06-01 15:48:29 -07:00

style.css

Update jax landing page

2024-08-20 13:03:08 -07:00

tensorboard_profiler.png

Add instructions for how to use the TensorBoard profiler to the profiling docs. (#3481)

2020-06-22 17:33:23 -07:00

type_lattice.svg

Fix nomenclature in type promotion doc

2022-02-15 13:26:06 -08:00

vscode-completion.png

JEP 12049: Type Annotation Roadmap

2022-09-13 09:14:48 -07:00

xla_spmd.jpg

added images for pjit tutorial

2021-09-08 23:29:20 +00:00