jax.profiler.start_server

jax.profiler.start_server(port)[source]

Starts a 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.

Returns a profiler server object. The server remains alive and listening until the server object is destroyed.

Parameters

port (int) –