jax.profiler.StepTraceContext
=============================

.. currentmodule:: jax.profiler

.. autoclass:: StepTraceContext

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~StepTraceContext.__init__
      ~StepTraceContext.is_enabled
      ~StepTraceContext.set_metadata