Toggle navigation

.rst .pdf

Warning

This page was created from a pull request (#9655).

jax.profiler.trace_function

jax.profiler.trace_function(*args, **kwargs)

previous

jax.profiler.save_device_memory_profile

next

jax.profiler.TraceContext