Templates

DGX Cloud Lepton provides two templates for you to get started quickly:

• MPI
• torchrun

To use the templates, you can click on the template dropdown button on the top right corner of the create job page.

MPI

Click on the MPI option, and you will see a modal pop up as shown below.

As you can see, you only need to fill in the run and setup fileds, and Lepton has already preset the necessary environment variables for you.

More information about mpirun, refer to the official documentation.

torchrun

Click on the touchrun option, and you will see a modal pop up as shown below.

Same as the MPI template, you only need to fill in the run and setup fileds, and Lepton has already preset the necessary environment variables for you.