Flyte 2 SDK

Reliably orchestrate ML pipelines, models, and agents at scale — in pure Python.

Install

pip install flyte

Example

import asyncio
import flyte

env = flyte.TaskEnvironment(
    name="hello_world",
    image=flyte.Image.from_debian_base(python_version=(3, 12)),
)

@env.task
def calculate(x: int) -> int:
    return x * 2 + 5

@env.task
async def main(numbers: list[int]) -> float:
    results = await asyncio.gather(*[
        calculate.aio(num) for num in numbers
    ])
    return sum(results) / len(results)

if __name__ == "__main__":
    flyte.init()
    run = flyte.run(main, numbers=list(range(10)))
    print(f"Result: {run.result}")

Python	Flyte CLI
python hello.py	flyte run hello.py main --numbers '[1,2,3]'

Serve a Model

# serving.py
from fastapi import FastAPI
import flyte
from flyte.app.extras import FastAPIAppEnvironment

app = FastAPI()
env = FastAPIAppEnvironment(
    name="my-model",
    app=app,
    image=flyte.Image.from_debian_base(python_version=(3, 12)).with_pip_packages(
        "fastapi", "uvicorn"
    ),
)

@app.get("/predict")
async def predict(x: float) -> dict:
    return {"result": x * 2 + 5}

if __name__ == "__main__":
    flyte.init_from_config()
    flyte.serve(env)

Python	Flyte CLI
python serving.py	flyte serve serving.py env

Local Development Experience

Install the TUI for a rich local development experience:

pip install flyte[tui]

Learn More

• Live Demo — Try Flyte 2 in your browser
• Documentation — Get started running locally
• SDK Reference — API reference docs
• CLI Reference — CLI docs
• Join the Flyte 2 Production Preview — Get early access
• Features — Async parallelism, app serving, tracing, and more
• Examples — Ready-to-run examples for every feature
• Contributing — Set up a dev environment and contribute
• Slack | GitHub Discussions | Issues

License

Apache 2.0 — see LICENSE.