nsys-jax: time-alignment and lock-step analysis - Githubissues

NVIDIA / JAX-Toolbox

JAX-Toolbox

Apache License 2.0

234 stars 44 forks source link

nsys-jax: time-alignment and lock-step analysis #926

Closed olupton closed 3 months ago

olupton commented 3 months ago

Also:

use "device" instead of "rank" everywhere
move warmup removal heuristics to standalone function