ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Set number of threads used within an individual op for parallelism.

Certain operations like matrix multiplication and reductions can utilize parallel threads for speed ups. A value of 0 means the system picks an appropriate number.

num_threads Number of parallel threads