Implement trace context manager and monkey-patch jax.profiler.trace in pathwaysutils.profiling#265
Closed
copybara-service[bot] wants to merge 1 commit into
Closed
Implement trace context manager and monkey-patch jax.profiler.trace in pathwaysutils.profiling#265copybara-service[bot] wants to merge 1 commit into
copybara-service[bot] wants to merge 1 commit into