Skip to main content

SkyPilot: An intercloud broker for the clouds

Project description

SkyPilot

Documentation GitHub Release Join Slack

Run AI on Any Infra — Unified, Faster, Cheaper


:fire: News :fire:

  • [Oct 2024] :tada: SkyPilot crossed 1M+ downloads :tada:: Thank you to our community! Twitter/X
  • [Sep 2024] Point, Launch and Serve Llama 3.2 on Kubernetes or Any Cloud: example
  • [Sep 2024] Run and deploy Pixtral, the first open-source multimodal model from Mistral AI.
  • [Jun 2024] Reproduce GPT with llm.c on any cloud: guide
  • [Apr 2024] Serve Qwen-110B on your infra: example
  • [Apr 2024] Using Ollama to deploy quantized LLMs on CPUs and GPUs: example
  • [Feb 2024] Deploying and scaling Gemma with SkyServe: example
  • [Feb 2024] Serving Code Llama 70B with vLLM and SkyServe: example
  • [Dec 2023] Mixtral 8x7B, a high quality sparse mixture-of-experts model, was released by Mistral AI! Deploy via SkyPilot on any cloud: example
  • [Nov 2023] Using Axolotl to finetune Mistral 7B on the cloud (on-demand and spot): example

LLM Finetuning Cookbooks: Finetuning Llama 2 / Llama 3.1 in your own cloud environment, privately: Llama 2 example and blog; Llama 3.1 example and blog

Archived

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:

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:

# Choose your clouds:
pip install -U "skypilot[kubernetes,aws,gcp,azure,oci,lambda,runpod,fluidstack,paperspace,cudo,ibm,scp]"

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

# Choose your clouds:
pip install "skypilot-nightly[kubernetes,aws,gcp,azure,oci,lambda,runpod,fluidstack,paperspace,cudo,ibm,scp]"

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

SkyPilot

Getting Started

You can find our documentation here.

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:

resources:
  accelerators: A100:8  # 8x NVIDIA A100 GPU

num_nodes: 1  # Number of VMs to launch

# Working directory (optional) containing the project codebase.
# Its contents are synced to ~/sky_workdir/ on the cluster.
workdir: ~/torch_examples

# Commands to be run before executing the job.
# Typical use: pip install -r requirements.txt, git clone, etc.
setup: |
  pip install "torch<2.2" torchvision --index-url https://download.pytorch.org/whl/cu121

# Commands to run as a job.
# Typical use: launch the main program.
run: |
  cd mnist
  python main.py --epochs 1

Prepare the workdir by cloning:

git clone https://github.com/pytorch/examples.git ~/torch_examples

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

sky launch my_task.yaml

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

SkyPilot Demo

Refer to Quickstart to get started with SkyPilot.

More Information

To learn more, see Concept: Sky Computing, SkyPilot docs, and SkyPilot blog.

Runnable examples:

Case Studies and Integrations: Community Spotlights

Follow updates:

Read the research:

SkyPilot was initially started at the Sky Computing Lab at UC Berkeley and has since gained many industry contributors. Read more about the project's origin here.

Support and Questions

We are excited to hear your feedback!

For general discussions, join us on the SkyPilot Slack.

Contributing

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

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

skypilot_nightly-1.0.0.dev20250118.tar.gz (954.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

skypilot_nightly-1.0.0.dev20250118-py3-none-any.whl (1.1 MB view details)

Uploaded Python 3

File details

Details for the file skypilot_nightly-1.0.0.dev20250118.tar.gz.

File metadata

File hashes

Hashes for skypilot_nightly-1.0.0.dev20250118.tar.gz
Algorithm Hash digest
SHA256 cc9d3c6dde584f57964a5b7a0ec1f6f575a2b027ee6ae6a4f8800af949438559
MD5 81d2e389858ece7ab15ccacec7cfc5e3
BLAKE2b-256 8c00132fa41aeaae1d98c05625f633fe52cee575b1aaf91b5edd41c5763c89c1

See more details on using hashes here.

File details

Details for the file skypilot_nightly-1.0.0.dev20250118-py3-none-any.whl.

File metadata

File hashes

Hashes for skypilot_nightly-1.0.0.dev20250118-py3-none-any.whl
Algorithm Hash digest
SHA256 a5e6a1cc3909219e7e519c3837446e110b2491dddb4d60a0c45d84a01cc8e207
MD5 1939f0fa1ad9d5075c8fd7526834e4d2
BLAKE2b-256 fe76a9698b40bb07e724bc6cd5b27ee794c9a456a2dbebec6a2cdca939f908bd

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page