Skip to main content

SkyPilot: An intercloud broker for the clouds

Project description

SkyPilot

Documentation GitHub Release Join Slack

Run LLMs and AI on Any Cloud


:fire: News :fire:

  • [Dec, 2023] Example: Using LoRAX to serve 1000s of finetuned LLMs on a single instance in the cloud: 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] Example: Using Axolotl to finetune Mistral 7B on the cloud (on-demand and spot): example
  • [Sep, 2023] Mistral 7B, a high-quality open LLM, was released! Deploy via SkyPilot on any cloud: Mistral docs
  • [Sep, 2023] Case study: Covariant transformed AI development on the cloud using SkyPilot, delivering models 4x faster cost-effectively: read the case study
  • [Aug, 2023] Cookbook: Finetuning Llama 2 in your own cloud environment, privately: example, blog post
  • [July, 2023] Self-Hosted Llama-2 Chatbot on Any Cloud: example
  • [June, 2023] Serving LLM 24x Faster On the Cloud with vLLM and SkyPilot: example, blog post
  • [April, 2023] SkyPilot YAMLs for finetuning & serving the Vicuna LLM with a single command!

SkyPilot is a framework for running LLMs, AI, and batch jobs on any cloud, offering maximum cost savings, highest GPU availability, and managed execution.

SkyPilot abstracts away cloud infra burdens:

  • Launch jobs & clusters on any cloud
  • Easy scale-out: queue and run many jobs, automatically managed
  • Easy access to object stores (S3, GCS, R2)

SkyPilot maximizes GPU availability for your jobs:

  • Provision in all zones/regions/clouds you have access to (the Sky), with automatic failover

SkyPilot cuts your cloud costs:

  • Managed Spot: 3-6x cost savings using spot VMs, with auto-recovery from preemptions
  • Optimizer: 2x cost savings by auto-picking the cheapest VM/zone/region/cloud
  • Autostop: hands-free cleanup of idle clusters

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

Install with pip (we recommend the nightly build for the latest features or from source):

pip install "skypilot-nightly[aws,gcp,azure,oci,lambda,runpod,ibm,scp,kubernetes]"  # choose your clouds

To get the last release, use:

pip install -U "skypilot[aws,gcp,azure,oci,lambda,runpod,ibm,scp,kubernetes]"  # choose your clouds

Current supported providers (AWS, Azure, GCP, OCI, Lambda Cloud, RunPod, IBM, Samsung, Cloudflare, any Kubernetes cluster):

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: V100:1  # 1x NVIDIA V100 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 torchvision

# 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 our Documentation and Tutorials.

Runnable examples:

Follow updates:

Read the research:

Support and Questions

We are excited to hear your feedback!

For general discussions, join us on the SkyPilot Slack.

Contributing

We welcome and value all contributions to the project! Please refer to 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.dev20240201.tar.gz (798.8 kB view details)

Uploaded Source

Built Distribution

File details

Details for the file skypilot-nightly-1.0.0.dev20240201.tar.gz.

File metadata

File hashes

Hashes for skypilot-nightly-1.0.0.dev20240201.tar.gz
Algorithm Hash digest
SHA256 1fd76caf7b58fd63b5f24478cb19a9fc491a26698e6e8d9d109556c104c3652f
MD5 5dbaa8782abc529f954a722e458de81c
BLAKE2b-256 841d4b2fb742e0e04451bdef6591ea74a56f468d2df341eeea58a7827ed9a2a6

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for skypilot_nightly-1.0.0.dev20240201-py3-none-any.whl
Algorithm Hash digest
SHA256 9c5add666ebf67771eec36a18d701b1fa9f57ed7f3bee2bed58a03c58f8cbff2
MD5 218c59a1684937694ca73d5f5e4653d7
BLAKE2b-256 4c8269595216867039fd58a1a3f063dbbe5e1d461445196917ae070e742de65e

See more details on using hashes here.

Supported by

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