mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/_static
History
barnesjoseph 81cdc882ae DOC: update main landing page style 
Co-authored-by: Jake VanderPlas <jakevdp@google.com>
2024-11-15 13:44:31 -08: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
jax-hero.svg
DOC: update main landing page style
2024-11-15 13:44:31 -08: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
DOC: update main landing page style
2024-11-15 13:44:31 -08: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