megatron.training.train\_step ============================= .. currentmodule:: megatron.training .. autofunction:: train_step