Pytorch Python Packages

huggingface-hub

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

transformers

🤗 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.

onnxruntime

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

datasets

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

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

thinc

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

sagemaker

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

wandb

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

ray

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

onnx

Open standard for machine learning interoperability

keras

Deep Learning for humans

torchmetrics

Machine learning metrics for distributed, scalable PyTorch applications.

timm

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

pytorch-lightning

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

peft

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

torchaudio

Data manipulation and transformation for audio signal processing, powered by PyTorch

diffusers

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

ultralytics

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

ultralytics-thop

PyTorch model profiler for computing MACs and parameter counts to measure deep learning model complexity.

lightning

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

tensorboardx

tensorboard for pytorch (and chainer, mxnet, numpy, ...)

bitsandbytes

Accessible large language models via k-bit quantization for PyTorch.

sagemaker-core

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

flashinfer-python

FlashInfer: Kernel Library for LLM Serving