Tensors and Dynamic neural networks in Python with strong GPU acceleration

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

Ultralytics YOLOv5 in PyTorch > ONNX > CoreML > TFLite

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

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

✅（已完结）超级全面的 深度学习 笔记【土堆 Pytorch】【李沐 动手学深度学习】【吴恩达 深度学习】【大飞 大模型Agent】

Datasets, Transforms and Models specific to Computer Vision

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

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

A PyTorch-based Speech Toolkit

Ultralytics YOLOv3 in PyTorch > ONNX > CoreML > TFLite

Deep universal probabilistic programming with Python and PyTorch