apex icon indicating copy to clipboard operation
apex copied to clipboard
verified publisher icon NVIDIA

Add BF16 support to FusedMixedPrecisionLamb

Open nv-joseli opened this issue 3 years ago • 0 comments

Added bfloat16 dispatch to kernels used in the FusedMixedPrecisionLamb optimizer.

nv-joseli avatar Jul 28 '22 19:07 nv-joseli