Check failed in collective_pipeliner when using gradient accumulation with non-unrolled loop - Githubissues

NVIDIA / JAX-Toolbox

JAX-Toolbox

Apache License 2.0

234 stars 44 forks source link

Check failed in collective_pipeliner when using gradient accumulation with non-unrolled loop #930

Closed qGentry closed 3 months ago

qGentry commented 3 months ago

/del