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

..

distributed_data_loading

Add Distributed data loading doc

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

pallas

[Pallas] Add Pallas design doc

2023-08-03 16:02:07 -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