#pytorch

AI Toolkit for Healthcare Imaging

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

Open standard for machine learning interoperability

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

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

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Neural Network Compression Framework for enhanced OpenVINO™ inference

A high-throughput and memory-efficient inference and serving engine for LLMs

LMCache: Supercharge Your LLM with the Fastest KV Cache Layer

Ongoing research training transformer models at scale

Scalable and user friendly neural :brain: forecasting algorithms.

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

Fast and Accurate ML in 3 Lines of Code

A library for training and deploying machine learning models on Amazon SageMaker

🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

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

The official CLI and Python client for the Hugging Face Hub.

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

Deep Learning for humans

On-device AI across mobile, embedded and edge for PyTorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Containers for machine learning

Ultralytics YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

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

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

FlagGems is an operator library for large language models implemented in the Triton Language.

FlashInfer: Kernel Library for LLM Serving

🐍 Geometric Computer Vision Library for Spatial AI

Standardized Distributed Generative and Predictive AI Inference Platform for Scalable, Multi-Framework Deployment on Kubernetes

PyTorch native quantization and sparsity for training and inference