still couldn’t resolve this issue, but I did get the amp_recipe.ipynb 22 to work with me. I get almost 2x speedup on RTX8000
same exact environment, the difference is that we don’t use apex in this code