jax.profiler.StepTraceAnnotation
================================

.. currentmodule:: jax.profiler

.. autoclass:: StepTraceAnnotation

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~StepTraceAnnotation.__init__
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~StepTraceAnnotation.is_enabled
      ~StepTraceAnnotation.set_metadata