jax authors
1bd781d992
Add JAX events that have time spans, not only durations.
...
Log such events for log_elapsed_time.
The rationale for not replacing durations with it is that it appears that
record_event_duration_secs() is widely used outside of the code of JAX itself.
PiperOrigin-RevId: 713167192
2025-01-07 23:08:30 -08:00
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
