Adding Onnx Support For Megatron-Bert (#2169)
* Adding Onnx Support For Megatron-Bert
* fix style
* fix style
* fix style
---------
Co-authored-by: pragyan <pragyan.singh@tarkas.ai>
Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>