SkyPilot

SkyPilot is a framework for running AI and batch workloads on any infra, offering unified execution, high cost savings, and high GPU availability.

SkyPilot abstracts away infra burdens:

• Launch dev clusters, jobs, and serving on any infra
• Easy job management: queue, run, and auto-recover many jobs

SkyPilot supports multiple clusters, clouds, and hardware (the Sky):

• Bring your reserved GPUs, Kubernetes clusters, or 12+ clouds
• Flexible provisioning of GPUs, TPUs, CPUs, with auto-retry

SkyPilot cuts your cloud costs & maximizes GPU availability:

• Autostop: automatic cleanup of idle resources
• Managed Spot: 3-6x cost savings using spot instances, with preemption auto-recovery
• Optimizer: 2x cost savings by auto-picking the cheapest & most available infra

SkyPilot supports your existing GPU, TPU, and CPU workloads, with no code changes.

Install with pip:

To get the latest features and fixes, use the nightly build or install from source:

Current supported infra (Kubernetes; AWS, GCP, Azure, OCI, Lambda Cloud, Fluidstack, RunPod, Cudo, Paperspace, Cloudflare, Samsung, IBM, VMware vSphere):

Getting Started

You can find our documentation here.

• Installation
• Quickstart
• CLI reference

SkyPilot in 1 Minute

A SkyPilot task specifies: resource requirements, data to be synced, setup commands, and the task commands.

Once written in this unified interface (YAML or Python API), the task can be launched on any available cloud. This avoids vendor lock-in, and allows easily moving jobs to a different provider.

Paste the following into a file my_task.yaml:

Prepare the workdir by cloning:

Launch with sky launch (note: access to GPU instances is needed for this example):

SkyPilot then performs the heavy-lifting for you, including:

1. Find the lowest priced VM instance type across different clouds
2. Provision the VM, with auto-failover if the cloud returned capacity errors
3. Sync the local workdir to the VM
4. Run the task's setup commands to prepare the VM for running the task
5. Run the task's run commands

Refer to Quickstart to get started with SkyPilot.

More Information

To learn more, see our documentation, blog, and community integrations.

Runnable examples:

• LLMs on SkyPilot
  • Llama 3.2: lightweight and vision models
  • Pixtral
  • Llama 3.1 finetuning and serving
  • GPT-2 via llm.c
  • Llama 3
  • Qwen
  • Databricks DBRX
  • Gemma
  • Mixtral 8x7B; Mistral 7B (from official Mistral team)
  • Code Llama
  • vLLM: Serving LLM 24x Faster On the Cloud (from official vLLM team)
  • SGLang: Fast and Expressive LLM Serving On the Cloud (from official SGLang team)
  • Vicuna chatbots: Training & Serving (from official Vicuna team)
  • Train your own Vicuna on Llama-2
  • Self-Hosted Llama-2 Chatbot
  • Ollama: Quantized LLMs on CPUs
  • LoRAX
  • QLoRA
  • LLaMA-LoRA-Tuner
  • Tabby: Self-hosted AI coding assistant
  • LocalGPT
  • Falcon
  • Add yours here & see more in llm/!
• Framework examples: PyTorch DDP, DeepSpeed, JAX/Flax on TPU, Stable Diffusion, Detectron2, Distributed TensorFlow, Ray Train, NeMo, programmatic grid search, Docker, Cog, Unsloth, Ollama, llm.c, Airflow and many more (examples/).

Case Studies and Integrations: Community Spotlights

Follow updates:

• Twitter
• Slack
• SkyPilot Blog (Introductory blog post)

Read the research:

• SkyPilot paper and talk (NSDI 2023)
• Sky Computing whitepaper
• Sky Computing vision paper (HotOS 2021)
• Policy for Managed Spot Jobs (NSDI 2024)

Support and Questions

We are excited to hear your feedback!

• For issues and feature requests, please open a GitHub issue.
• For questions, please use GitHub Discussions.

For general discussions, join us on the SkyPilot Slack.

Contributing

We welcome all contributions to the project! See CONTRIBUTING for how to get involved.