mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-25 03:26:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
22,072 Commits 508 Branches 263 Tags
Commit Graph

4 Commits

Author SHA1 Message Date
jax authors
fc1e1d4a65 Add freshness metablock to JAX OSS docs. 
PiperOrigin-RevId: 645508135
 2024-06-21 14:50:49 -07:00
Sergei Lebedev
01194bd2fb Clarified the type of the inputs to callback APIs 
The callback APIs were migrated to use jax.Arrays for both inputs and outputs
in JAX 0.4.27.

PiperOrigin-RevId: 634473890
 2024-05-16 11:29:09 -07:00
tempoxylophone
4bcad15111
Fix type in advanced-autodiff.md 
removed `doc.g` text from the first section.
 2024-05-14 11:52:41 -04:00
Jake VanderPlas
10ed827fe9 DOC: replace old tutorials with new content 2024-05-03 12:20:06 -07:00