Libraries & Tools

The model-definition framework for state-of-the-art ML models across text, vision, and audio.

Build and share machine learning demos and apps entirely in Python.

Open-source LLM engineering platform: tracing, evals, and observability for agents.

State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

Fast BPE tokenizer for use with OpenAI's models.