jax.profiler.start_server#

jax.profiler.start_server(port)[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) –