jax.profiler.annotate\_function =============================== .. currentmodule:: jax.profiler .. autofunction:: annotate_function