Pytorch GitHub Repositories (2026) · Open Source Tools

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

Tensor Fusion is a state-of-the-art GPU virtualization and pooling solution designed to optimize GPU cluster utilization to its fullest potential.

Generative Flow Networks - GFlowNet

Distributed Reinforcement Learning accelerated by Lightning Fabric

Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for interdisciplinary research, part of the 🔥PyTorch ecosystem. ⭐ Star to support our work!

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

We write your reusable computer vision tools. 💜

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

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

Real-time, local speech-to-text with streaming ASR, speaker diarization, translation, and OpenAI/Deepgram-compatible APIs.

Fast and Accurate ML in 3 Lines of Code

SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer

SD.Next: All-in-one WebUI for AI generative image and video creation, captioning and processing

Flower: A Friendly Federated AI Framework

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultralytics / MMEngine / Keras etc.

Scalable and user friendly neural :brain: forecasting algorithms.

A simple, high-quality voice conversion tool focused on ease of use and performance.

LakeSoul is an end-to-end, realtime cloud-native Lakehouse framework for fast data ingestion, concurrent updates, incremental analytics, multimodal data processing and vector search — powering next-generation BI and AI workloads.

Pytorch domain library for recommendation systems