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

3 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
Jake VanderPlas
10ed827fe9 DOC: replace old tutorials with new content 2024-05-03 12:20:06 -07:00