Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

A python library for user-friendly forecasting and anomaly detection on time series.

Deep universal probabilistic programming with Python and PyTorch

An Open Source Machine Learning Framework for Everyone

The fastai deep learning library

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

💫 Industrial-strength Natural Language Processing (NLP) in Python

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

Low-code framework for building custom LLMs, neural networks, and other AI models

The Triton Inference Server provides an optimized cloud and edge inferencing solution.

A Python library for anomaly detection across tabular, time series, graph, text, and image data. 60+ detectors, benchmark-backed ADEngine orchestration, and an agentic workflow for AI agents.