Current scaling: two-stage amax kernel#369

matthiasdiener wants to merge 15 commits intoROCm:devROCm/TransformerEngine:devfrom

matthiasdiener:speedup-amax-kernelmatthiasdiener/TransformerEngine:speedup-amax-kernelCopy head branch name to clipboard