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