mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,443 Commits 501 Branches 263 Tags
Commit Graph

6 Commits

Author SHA1 Message Date
Jane Liu
522a8fd792 consolidate the code example 2024-12-19 14:47:29 -08:00
Jane Liu
f124232f53 Add activation offloading examples in a subsection 2024-12-19 14:47:29 -08:00
nireekshak
f43fa9fc78 Fix some typos 2024-12-03 05:07:53 +00:00
Frédéric Bastien
e9011940d8
Update docs/gradient-checkpointing.md 
Co-authored-by: Matthew Johnson <mattjj@google.com>
 2024-10-11 12:33:10 -04:00
Frederic Bastien
a159c0f417 Document jax.checkpoint policies. 2024-09-22 16:05:20 -04:00
8bitmp3
0cf040c9a1 Add/update JAX Advanced Tutorials docs, ToC structure 2024-09-20 23:06:54 +00:00