pyt21.11 compatibility: (1) enable nvfuser for bf16 (2) change of api…
The current NeMo code will crash at PyTorch21.11. 
This update will crash at PyTorch21.10 so should be made sure this is packaged with PyTorch21.11.

Signed-off-by: Sangkug Lym <slym@nvidia.com>