* [docker] megatron: add TE to ngc dockerfile

* fix fa

* add py-spy

* set val_before_train=False for megatron