lightning-sdk 2025.9.23__py3-none-any.whl → 2025.9.29__py3-none-any.whl

lightning_sdk/__init__.py

@@ -34,6 +34,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.09.23"
+__version__ = "2025.09.29"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/cli/entrypoint.py

@@ -21,6 +21,7 @@ from lightning_sdk.cli.groups import (
     # job,
     # mmt,
     studio,
+    vm,
 )
 from lightning_sdk.cli.utils import CustomHelpFormatter, rich_to_str
 from lightning_sdk.constants import _LIGHTNING_DEBUG
@@ -40,7 +41,7 @@ def _notify_exception(exception_type: Type[BaseException], value: BaseException,
     if _LIGHTNING_DEBUG:
         tb_text = "".join(traceback.format_exception(exception_type, value, tb))
         renderables.append(Text("\n\nFull traceback:\n", style="bold yellow"))
-        renderables.append(Syntax(tb_text, "python", theme="monokai", line_numbers=False, word_wrap=True))
+        renderables.append(Syntax(tb_text, "python", theme="monokai light", line_numbers=False, word_wrap=True))
     else:
         renderables.append(Text("\n\n🐞 To view the full traceback, set: LIGHTNING_DEBUG=1"))
 
@@ -83,6 +84,7 @@ main_cli.add_command(config)
 # main_cli.add_command(job)
 # main_cli.add_command(mmt)
 main_cli.add_command(studio)
+main_cli.add_command(vm)
 if os.environ.get("LIGHTNING_EXPERIMENTAL_CLI_ONLY", "0") != "1":
     #### LEGACY COMMANDS ####
     # these commands are currently supported for backwards compatibility, but will potentially be removed in the future.

lightning_sdk/cli/groups.py

@@ -6,6 +6,7 @@ from lightning_sdk.cli.config import register_commands as register_config_comman
 from lightning_sdk.cli.job import register_commands as register_job_commands
 from lightning_sdk.cli.mmt import register_commands as register_mmt_commands
 from lightning_sdk.cli.studio import register_commands as register_studio_commands
+from lightning_sdk.cli.vm import register_commands as register_vm_commands
 
 
 @click.group(name="studio")
@@ -25,7 +26,12 @@ def mmt() -> None:
 
 @click.group(name="config")
 def config() -> None:
-    """Manage Lightning SDK and CLIconfiguration."""
+    """Manage Lightning SDK and CLI configuration."""
+
+
+@click.group(name="vm")
+def vm() -> None:
+    """Manage Lightning AI VMs."""
 
 
 # Register config commands with the main config group
@@ -33,3 +39,4 @@ register_job_commands(job)
 register_mmt_commands(mmt)
 register_studio_commands(studio)
 register_config_commands(config)
+register_vm_commands(vm)

lightning_sdk/cli/legacy/entrypoint.py

@@ -48,7 +48,7 @@ def _notify_exception(exception_type: Type[BaseException], value: BaseException,
     if _LIGHTNING_DEBUG:
         tb_text = "".join(traceback.format_exception(exception_type, value, tb))
         renderables.append(Text("\n\nFull traceback:\n", style="bold yellow"))
-        renderables.append(Syntax(tb_text, "python", theme="monokai", line_numbers=False, word_wrap=True))
+        renderables.append(Syntax(tb_text, "python", theme="monokai light", line_numbers=False, word_wrap=True))
     else:
         renderables.append(Text("\n\n🐞 To view the full traceback, set: LIGHTNING_DEBUG=1"))
 

lightning_sdk/cli/studio/create.py

@@ -9,7 +9,7 @@ from lightning_sdk.cli.utils.save_to_config import save_teamspace_to_config
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.machine import CloudProvider
-from lightning_sdk.studio import Studio
+from lightning_sdk.studio import VM, Studio
 
 
 @click.command("create")
@@ -36,6 +36,16 @@ def create_studio(
     Example:
         lightning studio create
     """
+    create_impl(name=name, teamspace=teamspace, cloud_provider=cloud_provider, cloud_account=cloud_account, vm=False)
+
+
+def create_impl(
+    name: Optional[str],
+    teamspace: Optional[str],
+    cloud_provider: Optional[str],
+    cloud_account: Optional[str],
+    vm: bool,
+) -> None:
     menu = TeamspacesMenu()
 
     resolved_teamspace = menu(teamspace)
@@ -44,8 +54,12 @@ def create_studio(
     if cloud_provider is not None:
         cloud_provider = CloudProvider(cloud_provider)
 
+    create_cls = VM if vm else Studio
+    cls_name = create_cls.__qualname__
+
     try:
-        studio = Studio(
+        create_cls = VM if vm else Studio
+        studio = create_cls(
             name=name,
             teamspace=resolved_teamspace,
             create_ok=True,
@@ -54,7 +68,7 @@ def create_studio(
         )
     except (RuntimeError, ValueError, ApiException):
         if name:
-            raise ValueError(f"Could not create Studio: '{name}'. Does the Studio exist?") from None
-        raise ValueError(f"Could not create Studio: '{name}'. Please provide a Studio name") from None
+            raise ValueError(f"Could not create {cls_name}: '{name}'. Does the {cls_name} exist?") from None
+        raise ValueError(f"Could not create {cls_name}: '{name}'. Please provide a {cls_name} name") from None
 
-    click.echo(f"Studio {studio_name_link(studio)} created successfully")
+    click.echo(f"{cls_name} {studio_name_link(studio)} created successfully")

lightning_sdk/cli/studio/delete.py

@@ -12,7 +12,7 @@ from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 @click.option(
     "--name",
     help=(
-        "The name of the studio to start. "
+        "The name of the studio to delete. "
         "If not provided, will try to infer from environment, "
         "use the default value from the config or prompt for interactive selection."
     ),
@@ -25,21 +25,25 @@ def delete_studio(name: Optional[str] = None, teamspace: Optional[str] = None) -
       lightning studio delete --name my-studio
 
     """
+    return delete_impl(name=name, teamspace=teamspace, vm=False)
+
+
+def delete_impl(name: Optional[str], teamspace: Optional[str], vm: bool) -> None:
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
-    menu = StudiosMenu(resolved_teamspace)
+    menu = StudiosMenu(resolved_teamspace, vm=vm)
     studio = menu(studio=name)
 
     studio_name = f"{studio.teamspace.owner.name}/{studio.teamspace.name}/{studio.name}"
     confirmed = click.confirm(
-        f"Are you sure you want to delete studio '{studio_name}'?",
+        f"Are you sure you want to delete {studio._cls_name} '{studio_name}'?",
         abort=True,
     )
     if not confirmed:
-        click.echo("Studio deletion cancelled")
+        click.echo(f"{studio._cls_name} deletion cancelled")
         return
 
     studio.delete()
 
-    click.echo(f"Studio '{studio.name}' deleted successfully")
+    click.echo(f"{studio._cls_name} '{studio.name}' deleted successfully")

lightning_sdk/cli/studio/list.py

@@ -35,13 +35,17 @@ def list_studios(teamspace: Optional[str] = None, all: bool = False, sort_by: Op
         lightning studio list --teamspace owner/teamspace
 
     """
+    return list_impl(teamspace=teamspace, all=all, sort_by=sort_by, vm=False)
+
+
+def list_impl(teamspace: Optional[str], all: bool, sort_by: Optional[str], vm: bool) -> None:  # noqa: A002
     menu = TeamspacesMenu()
     teamspace_resolved = menu(teamspace=teamspace)
     save_teamspace_to_config(teamspace_resolved, overwrite=False)
 
     user = _get_authed_user()
 
-    studios = teamspace_resolved.studios
+    studios = teamspace_resolved.vms if vm else teamspace_resolved.studios
 
     table = Table(
         pad_edge=True,

lightning_sdk/cli/studio/ssh.py

@@ -20,7 +20,7 @@ from lightning_sdk.utils.config import _DEFAULT_CONFIG_FILE_PATH
 @click.option(
     "--name",
     help=(
-        "The name of the studio to start. "
+        "The name of the studio to ssh into. "
         "If not provided, will try to infer from environment, "
         "use the default value from the config or prompt for interactive selection."
     ),
@@ -39,6 +39,10 @@ def ssh_studio(name: Optional[str] = None, teamspace: Optional[str] = None, opti
     Example:
         lightning studio ssh --name my-studio
     """
+    return ssh_impl(name=name, teamspace=teamspace, option=option, vm=False)
+
+
+def ssh_impl(name: Optional[str], teamspace: Optional[str], option: Optional[List[str]], vm: bool) -> None:
     auth = Auth()
     auth.authenticate()
     ssh_private_key_path = _download_ssh_keys(auth.api_key, force_download=False)
@@ -46,8 +50,10 @@ def ssh_studio(name: Optional[str] = None, teamspace: Optional[str] = None, opti
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
-    menu = StudiosMenu(resolved_teamspace)
-    studio = menu(studio=name)
+    menu = StudiosMenu(resolved_teamspace, vm=vm)
+    studio = menu(
+        studio=name,
+    )
     save_studio_to_config(studio)
 
     ssh_options = " -o " + " -o ".join(option) if option else ""

lightning_sdk/cli/studio/start.py

@@ -9,7 +9,7 @@ from lightning_sdk.cli.utils.save_to_config import save_studio_to_config
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.machine import CloudProvider, Machine
-from lightning_sdk.studio import Studio
+from lightning_sdk.studio import VM, Studio
 
 
 @click.command("start")
@@ -57,6 +57,28 @@ def start_studio(
         lightning studio start --name my-studio
 
     """
+    return start_impl(
+        name=name,
+        teamspace=teamspace,
+        create=create,
+        machine=machine,
+        interruptible=interruptible,
+        cloud_provider=cloud_provider,
+        cloud_account=cloud_account,
+        vm=False,
+    )
+
+
+def start_impl(
+    name: Optional[str],
+    teamspace: Optional[str],
+    create: bool,
+    machine: str,
+    interruptible: bool,
+    cloud_provider: Optional[str],
+    cloud_account: Optional[str],
+    vm: bool,
+) -> None:
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
@@ -64,10 +86,11 @@ def start_studio(
         cloud_provider = CloudProvider(cloud_provider)
 
     if not create:
-        menu = StudiosMenu(resolved_teamspace)
+        menu = StudiosMenu(resolved_teamspace, vm=vm)
         studio = menu(studio=name)
     else:
-        studio = Studio(
+        create_cls = VM if vm else Studio
+        studio = create_cls(
             name=name,
             teamspace=resolved_teamspace,
             create_ok=create,

lightning_sdk/cli/studio/stop.py

@@ -14,7 +14,7 @@ from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 @click.option(
     "--name",
     help=(
-        "The name of the studio to start. "
+        "The name of the studio to stop. "
         "If not provided, will try to infer from environment, "
         "use the default value from the config or prompt for interactive selection."
     ),
@@ -27,15 +27,19 @@ def stop_studio(name: Optional[str] = None, teamspace: Optional[str] = None) ->
         lightning studio stop --name my-studio
 
     """
+    return stop_impl(name=name, teamspace=teamspace, vm=False)
+
+
+def stop_impl(name: Optional[str], teamspace: Optional[str], vm: bool) -> None:
     # missing studio_name and teamspace are handled by the studio class
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
-    menu = StudiosMenu(resolved_teamspace)
+    menu = StudiosMenu(resolved_teamspace, vm=vm)
     studio = menu(studio=name)
 
     studio.stop()
 
     save_studio_to_config(studio)
 
-    click.echo(f"Studio {studio_name_link(studio)} stopped successfully")
+    click.echo(f"{studio._cls_name} {studio_name_link(studio)} stopped successfully")

lightning_sdk/cli/studio/switch.py

@@ -9,14 +9,13 @@ from lightning_sdk.cli.utils.save_to_config import save_studio_to_config
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.machine import Machine
-from lightning_sdk.studio import Studio
 
 
 @click.command("switch")
 @click.option(
     "--name",
     help=(
-        "The name of the studio to start. "
+        "The name of the studio to switch to a different machine. "
         "If not provided, will try to infer from environment, "
         "use the default value from the config or prompt for interactive selection."
     ),
@@ -35,16 +34,33 @@ def switch_studio(
     interruptible: bool = False,
 ) -> None:
     """Switch a Studio to a different machine type."""
+    return switch_impl(
+        name=name,
+        teamspace=teamspace,
+        machine=machine,
+        interruptible=interruptible,
+        vm=False,
+    )
+
+
+def switch_impl(
+    name: Optional[str],
+    teamspace: Optional[str],
+    machine: Optional[str],
+    interruptible: bool,
+    vm: bool,
+) -> None:
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
-    menu = StudiosMenu(resolved_teamspace)
+    menu = StudiosMenu(resolved_teamspace, vm=vm)
     studio = menu(studio=name)
 
     resolved_machine = Machine.from_str(machine)
-    Studio.show_progress = True
+
+    studio.__class__.show_progress = True
     studio.switch_machine(resolved_machine, interruptible=interruptible)
 
     save_studio_to_config(studio)
 
-    click.echo(f"Studio {studio_name_link(studio)} switched to machine '{resolved_machine}' successfully")
+    click.echo(f"{studio._cls_name} {studio_name_link(studio)} switched to machine '{resolved_machine}' successfully")

lightning_sdk/cli/utils/studio_selection.py

@@ -1,12 +1,12 @@
 import os
 from contextlib import suppress
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, Union
 
 import click
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk.cli.legacy.exceptions import StudioCliError
-from lightning_sdk.studio import Studio
+from lightning_sdk.studio import VM, Studio
 from lightning_sdk.teamspace import Teamspace
 from lightning_sdk.utils.resolve import _get_authed_user
 
@@ -17,54 +17,58 @@ class StudiosMenu:
     It can be used to select a studio from a list of possible studios, or to resolve a studio from a name.
     """
 
-    def __init__(self, teamspace: Teamspace) -> None:
+    def __init__(self, teamspace: Teamspace, vm: bool = False) -> None:
         """Initialize the StudiosMenu with a teamspace.
 
         Args:
             teamspace: The teamspace to list studios from
         """
         self.teamspace = teamspace
+        self.vm = vm
 
-    def _get_studio_from_interactive_menu(self, possible_studios: Dict[str, Studio]) -> Studio:
+    def _get_studio_from_interactive_menu(self, possible_studios: Dict[str, Union[Studio, VM]]) -> Union[Studio, VM]:
         studio_names = sorted(possible_studios.keys())
-        terminal_menu = self._prepare_terminal_menu_studios(studio_names)
+        terminal_menu = self._prepare_terminal_menu_studios(studio_names, vm=self.vm)
         terminal_menu.show()
 
         selected_name = studio_names[terminal_menu.chosen_menu_index]
         return possible_studios[selected_name]
 
-    def _get_studio_from_name(self, studio: str, possible_studios: Dict[str, Studio]) -> Studio:
+    def _get_studio_from_name(self, studio: str, possible_studios: Dict[str, Union[Studio, VM]]) -> Union[Studio, VM]:
         if studio in possible_studios:
             return possible_studios[studio]
 
-        click.echo(f"Could not find Studio {studio}, please select it from the list:")
+        click.echo(f"Could not find {'VM' if self.vm else 'Studio'} {studio}, please select it from the list:")
         return self._get_studio_from_interactive_menu(possible_studios)
 
     @staticmethod
-    def _prepare_terminal_menu_studios(studio_names: List[str], title: Optional[str] = None) -> TerminalMenu:
+    def _prepare_terminal_menu_studios(
+        studio_names: List[str], title: Optional[str] = None, vm: bool = False
+    ) -> TerminalMenu:
         if title is None:
-            title = "Please select a Studio out of the following:"
+            title = f"Please select a {'VM' if vm else 'Studio'} out of the following:"
 
         return TerminalMenu(studio_names, title=title, clear_menu_on_exit=True)
 
-    def _get_possible_studios(self) -> Dict[str, Studio]:
+    def _get_possible_studios(self) -> Dict[str, Union[Studio, VM]]:
         """Get all available studios in the teamspace."""
         studios = {}
 
         user = _get_authed_user()
+        studios = self.teamspace.vms if self.vm else self.teamspace.studios
         for studio in self.teamspace.studios:
             if studio._studio.user_id == user.id:
                 studios[studio.name] = studio
         return studios
 
-    def __call__(self, studio: Optional[str] = None) -> Studio:
+    def __call__(self, studio: Optional[str] = None) -> Union[Studio, VM]:
         """Select a studio from the teamspace.
 
         Args:
             studio: Optional studio name to select. If not provided, will show interactive menu.
 
         Returns:
-            Selected Studio object
+            Selected Studio/VM object
 
         Raises:
             StudioCliError: If studio selection fails
@@ -73,15 +77,18 @@ class StudiosMenu:
             # try to resolve the studio from the name, environment or config
             resolved_studio = None
 
+            selected_cls = VM if self.vm else Studio
+
             with suppress(Exception):
-                resolved_studio = Studio(name=studio, teamspace=self.teamspace, create_ok=False)
+                resolved_studio = selected_cls(name=studio, teamspace=self.teamspace, create_ok=False)
 
             if resolved_studio is not None:
                 return resolved_studio
 
             if os.environ.get("LIGHTNING_NON_INTERACTIVE", "0") == "1" and studio is None:
                 raise ValueError(
-                    "Studio selection is not supported in non-interactive mode. Please provide a studio name."
+                    f"{'VM' if self.vm else 'Studio'} selection is not supported in non-interactive mode. "
+                    "Please provide a studio name."
                 )
 
             click.echo(f"Listing studios in teamspace {self.teamspace.owner.name}/{self.teamspace.name}...")
@@ -101,6 +108,6 @@ class StudiosMenu:
 
         except Exception as e:
             raise StudioCliError(
-                "Could not resolve a Studio. "
+                f"Could not resolve a {'VM' if self.vm else 'Studio'}. "
                 "Please pass it as an argument or contact Lightning AI directly to resolve this issue."
             ) from e

lightning_sdk/cli/vm/__init__.py

@@ -0,0 +1,20 @@
+import click
+
+
+def register_commands(group: click.Group) -> None:
+    """Register studio commands with the given group."""
+    from lightning_sdk.cli.vm.create import create_vm
+    from lightning_sdk.cli.vm.delete import delete_vm
+    from lightning_sdk.cli.vm.list import list_vms
+    from lightning_sdk.cli.vm.ssh import ssh_vm
+    from lightning_sdk.cli.vm.start import start_vm
+    from lightning_sdk.cli.vm.stop import stop_vm
+    from lightning_sdk.cli.vm.switch import switch_vm
+
+    group.add_command(create_vm)
+    group.add_command(delete_vm)
+    group.add_command(list_vms)
+    group.add_command(ssh_vm)
+    group.add_command(start_vm)
+    group.add_command(stop_vm)
+    group.add_command(switch_vm)

lightning_sdk/cli/vm/create.py

@@ -0,0 +1,33 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.create import create_impl
+from lightning_sdk.machine import CloudProvider
+
+
+@click.command("create")
+@click.option("--name", help="The name of the VM to create. If not provided, a random name will be generated.")
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option(
+    "--cloud-provider",
+    help="The cloud provider to start the VM on. Defaults to teamspace default.",
+    type=click.Choice(m.name for m in list(CloudProvider)),
+)
+@click.option(
+    "--cloud-account",
+    help="The cloud account to create the VM on. Defaults to teamspace default.",
+    type=click.STRING,
+)
+def create_vm(
+    name: Optional[str] = None,
+    teamspace: Optional[str] = None,
+    cloud_provider: Optional[str] = None,
+    cloud_account: Optional[str] = None,
+) -> None:
+    """Create a new VM.
+
+    Example:
+        lightning vm create
+    """
+    create_impl(name=name, teamspace=teamspace, cloud_provider=cloud_provider, cloud_account=cloud_account, vm=True)

lightning_sdk/cli/vm/delete.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.delete import delete_impl
+
+
+@click.command("delete")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to delete. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+def delete_vm(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Delete a VM.
+
+    Example:
+      lightning vm delete --name my-vm
+
+    """
+    return delete_impl(name=name, teamspace=teamspace, vm=True)

lightning_sdk/cli/vm/list.py

@@ -0,0 +1,30 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.list import list_impl
+
+
+@click.command("list")
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option(
+    "--all",
+    is_flag=True,
+    flag_value=True,
+    default=False,
+    help="List all VMs, not just the ones belonging to the authed user",
+)
+@click.option(
+    "--sort-by",
+    default=None,
+    type=click.Choice(["name", "teamspace", "status", "machine", "cloud-account"], case_sensitive=False),
+    help="the attribute to sort the VMs by.",
+)
+def list_vms(teamspace: Optional[str] = None, all: bool = False, sort_by: Optional[str] = None) -> None:  # noqa: A002
+    """List VMs in a teamspace.
+
+    Example:
+        lightning vm list --teamspace owner/teamspace
+
+    """
+    return list_impl(teamspace=teamspace, all=all, sort_by=sort_by, vm=True)

lightning_sdk/cli/vm/ssh.py

@@ -0,0 +1,31 @@
+from typing import List, Optional
+
+import click
+
+from lightning_sdk.cli.studio.ssh import ssh_impl
+
+
+@click.command("ssh")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to ssh into. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)", type=click.STRING)
+@click.option(
+    "--option",
+    "-o",
+    help="Additional options to pass to the SSH command. Can be specified multiple times.",
+    multiple=True,
+    type=click.STRING,
+)
+def ssh_vm(name: Optional[str] = None, teamspace: Optional[str] = None, option: Optional[List[str]] = None) -> None:
+    """SSH into a VM.
+
+    Example:
+        lightning vm ssh --name my-vm
+    """
+    return ssh_impl(name=name, teamspace=teamspace, option=option, vm=False)

lightning_sdk/cli/vm/start.py

@@ -0,0 +1,60 @@
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.studio.start import start_impl
+from lightning_sdk.machine import CloudProvider, Machine
+
+
+@click.command("start")
+@click.option(
+    "--name",
+    help=(
+        "The name of the VM to start. "
+        "If not provided, will try to infer from environment, "
+        "use the default value from the config or prompt for interactive selection."
+    ),
+)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option("--create", is_flag=True, help="Create the VM if it doesn't exist")
+@click.option(
+    "--machine",
+    help="The machine type to start the VM on. Defaults to CPU-4",
+    type=click.Choice(m.name for m in Machine.__dict__.values() if isinstance(m, Machine) and m._include_in_cli),
+)
+@click.option("--interruptible", is_flag=True, help="Start the VM on an interruptible instance.")
+@click.option(
+    "--cloud-provider",
+    help=("The cloud provider to start the VM on. Defaults to teamspace default. Only used if --create is specified."),
+    type=click.Choice(m.name for m in list(CloudProvider)),
+)
+@click.option(
+    "--cloud-account",
+    help="The cloud account to start the VM on. Defaults to teamspace default. Only used if --create is specified.",
+    type=click.STRING,
+)
+def start_vm(
+    name: Optional[str] = None,
+    teamspace: Optional[str] = None,
+    create: bool = False,
+    machine: str = "CPU",
+    interruptible: bool = False,
+    cloud_provider: Optional[str] = None,
+    cloud_account: Optional[str] = None,
+) -> None:
+    """Start a VM.
+
+    Example:
+        lightning vm start --name my-vm
+
+    """
+    return start_impl(
+        name=name,
+        teamspace=teamspace,
+        create=create,
+        machine=machine,
+        interruptible=interruptible,
+        cloud_provider=cloud_provider,
+        cloud_account=cloud_account,
+        vm=True,
+    )