To upgrade an already setup environment: ```bash cd run_scripts source variables.bash ln -s ../fixed_torch_run.py "$MEGATRON_DEEPSPEED_REPO" ```