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

4 Commits

Author SHA1 Message Date
jax authors
be7f210a1b Allow duration event listeners to take extra keyword arguments. 
PiperOrigin-RevId: 570783105
 2023-10-04 13:15:49 -07:00
Chansoo Lee
79d0a83069 Allow event listners to take extra keyword arguments. 
PiperOrigin-RevId: 569138957
 2023-09-28 04:38:43 -07:00
Peter Hawkins
a28b01243b Move contents of jax.monitoring to jax._src.monitoring. 
PiperOrigin-RevId: 508191560
 2023-02-08 15:03:22 -08:00
jax authors
726b2bc2ee Add JAX monitoring library that instruments code via events. 
PiperOrigin-RevId: 488731805
 2022-11-15 12:41:41 -08:00