jax.profiler.annotate\_function
===============================

.. currentmodule:: jax.profiler

.. autofunction:: annotate_function