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.

This repository showcases various advanced techniques for Retrieval-Augmented Generation (RAG) systems. Each technique has a detailed notebook tutorial.

✅（已完结）超级全面的 深度学习 笔记【土堆 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.

Pocket Flow: Codebase to Tutorial

A PyTorch-based Speech Toolkit