jax.profiler.trace\_function
============================

.. currentmodule:: jax.profiler

.. autofunction:: trace_function