skypilot-nightly

Pre-release

This release is a pre-release and may not be stable for production use.

SkyPilot

Run LLMs and AI on Any Cloud

:fire: News :fire:

[Aug, 2023] New 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
[June, 2023] Two new clouds supported: Samsung SCP and Oracle OCI!
[April, 2023] SkyPilot YAMLs released for finetuning & serving the Vicuna model with a single command!
[March, 2023] Vicuna LLM chatbot trained using SkyPilot for $300 on spot instances!

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:

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

To get the latest features/updates, install from source or the nightly build:

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

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

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:

Find the lowest priced VM instance type across different clouds
Provision the VM, with auto-failover if the cloud returned capacity errors
Sync the local workdir to the VM
Run the task's setup commands to prepare the VM for running the task
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:

LLMs on SkyPilot
- Train Your Own Vicuna on Llama-2
- Self-Hosted Llama-2 Chatbot
- Vicuna chatbots: Training & Serving (from official Vicuna team)
- vLLM: Serving LLM 24x Faster On the Cloud (from official vLLM team)
- QLoRA
- LLaMA-LoRA-Tuner
- Tabby: Self-hosted AI coding assistant
- LocalGPT
- Add yours here & see more in llm/!
Framework examples: PyTorch DDP, DeepSpeed, JAX/Flax on TPU, Stable Diffusion, Detectron2, Distributed TensorFlow, programmatic grid search, Docker, and many more (examples/).

Follow updates:

Read the research:

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 and value all contributions to the project! Please refer to CONTRIBUTING for how to get involved.

Metadata

Release files for skypilot-nightly 1.0.0.dev20230921

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for skypilot-nightly 1.0.0.dev20230921
File	Size	Uploaded
skypilot-nightly-1.0.0.dev20230921.tar.gz	641.6 kB	Sep 21, 2023	Details

Built distribution (wheel)

Table of built distributions (wheels) for skypilot-nightly 1.0.0.dev20230921
File	Interpreter	ABI	Platform	Reset
skypilot_nightly-1.0.0.dev20230921-py3-none-any.whl 705.4 kB Sep 21, 2023	Python 3	none	any	Details

Total release size: 1.3 MB

Release files / skypilot-nightly-1.0.0.dev20230921.tar.gz

Download URL	skypilot-nightly-1.0.0.dev20230921.tar.gz
Size	641.6 kB
Tags	Source
SHA-256 checksum How to use checksums	`497bd1ae101cc494668e2e97e711c99f928fa33a7d307fd4b27fa765c369e425`
BLAKE2b-256 checksum How to use checksums	`5e8bbaa142bdafc5814155432a3279198a6eba04d95d220b8d81d42bd33ef9bb`
Upload date	Sep 21, 2023
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/4.0.2 CPython/3.11.5`

Release files / skypilot_nightly-1.0.0.dev20230921-py3-none-any.whl

Download URL	skypilot_nightly-1.0.0.dev20230921-py3-none-any.whl
Size	705.4 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`a50ef6639fbfed44c46715d91a3cc88ed6b9fb14c9249957165c999ac1771c1f`
BLAKE2b-256 checksum How to use checksums	`18758848e205ec6424c4201cec521f89885918d497c9acd5dbaa319509f662a6`
Upload date	Sep 21, 2023
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/4.0.2 CPython/3.11.5`