mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 22:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
17,470 Commits 508 Branches 263 Tags
Commit Graph

18 Commits

Author SHA1 Message Date
Jake VanderPlas
e21286d6f0 JEP 9263: Typed keys & pluggable RNGs 
Co-authored-by: Roy Frostig <frostig@google.com>
 2023-09-20 11:26:13 -07:00
Sharad Vikram
96e2d93f53 [Pallas] Add Pallas design doc 2023-08-03 16:02:07 -07:00
Sharad Vikram
7f1ef32ba3 Add initial documentation for Pallas 2023-08-03 12:30:19 -07:00
Jake VanderPlas
55d9c06267 DOC: update sphinx & sphinx-autodoc-typehints 2023-03-01 11:03:20 -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
Jake VanderPlas
358363e17f JEP 12049: Type Annotation Roadmap 2022-09-13 09:14:48 -07:00
Sharad Vikram
4386a0f909 Add debugging tools under jax.debug and documentation 
Co-authored-by: Matthew Johnson <mattjj@google.com>
Co-authored-by: Lena Martens <lenamartens@google.com>
 2022-07-28 20:07:26 -07:00
Sharad Vikram
76669835ba Add an option to create a perfetto link in the JAX profiler 2022-06-01 15:48:29 -07:00
Jake VanderPlas
e82f232ea9 Fix nomenclature in type promotion doc 2022-02-15 13:26:06 -08:00
Tina Jia
add8300372 added images for pjit tutorial 2021-09-08 23:29:20 +00:00
jax authors
860d1b3de4 Merge pull request #4815 from j-towns:docs-favicon 
PiperOrigin-RevId: 341325726
 2020-11-08 19:06:24 -08:00
Jamie Townsend
46ca7a00ce Rm gridlines from favicon (thanks @shoyer) 2020-11-08 19:12:40 +00:00
Jake VanderPlas
86f4535393 DOC: add lattice to type promotion doc 2020-11-06 09:13:28 -08:00
Jamie Townsend
62683959b1 [DOCS] Add JAX favicon to sphinx 2020-11-06 13:43:18 +00:00
Peter Hawkins
5116fd47aa
Add a heap profiler API and document it. (#3576) 2020-06-26 17:09:09 -04:00
Skye Wanderman-Milne
8fe6da08d4
Add instructions for how to use the TensorBoard profiler to the profiling docs. (#3481) 2020-06-22 17:33:23 -07:00
Stephan Hoyer
515e07bdf2 White background 2019-08-05 13:08:17 -07:00
Stephan Hoyer
8c8222a319 Add the JAX logo to sphinx docs 2019-08-05 12:34:29 -07:00