jax
e073e254 - Add a jax.profiler.StepTraceContext API. (#3591)

Commit

5 years ago

Add a jax.profiler.StepTraceContext API. (#3591) * Add RootTraceContext. * Rename to StepTraceContext.

References

Author

Jiho Choi

Parents