megatron.checkpointing.save_checkpoint# megatron.checkpointing.save_checkpoint(iteration, model, optimizer, opt_param_scheduler)# Save a model checkpoint.