jax.profiler.start_server

Contents

jax.profiler.start_server#

jax.profiler.start_server(port, requires_backend=True)[source]#

Starts the profiler server on port port.

Using the “TensorFlow profiler” feature in TensorBoard 2.2 or newer, you can connect to the profiler server and sample execution traces that show CPU, GPU, and/or TPU device activity.

Parameters:

  • port (int) – The port to start the profiler server on.

  • requires_backend (bool) – If False, the profiler server will not wait for backends to be initialized before starting. Default is True.

Return type:

_profiler.ProfilerServer

Contents