Justin Fu bdb03309b5

Merge branch 'main' into pallas_distr_docs

2024-08-08 17:36:36 -07:00

..

distributed_data_loading

Add Distributed data loading doc

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

pallas

[Pallas] Add pallas distributed computation tutorial

2024-08-08 17:34:35 -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

[shape_poly] Add documentation for shape polymorphism

2024-06-15 18:20:54 +03: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