Find trending repositories by name or description.
⚡️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.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Implement a ChatGPT-like LLM in PyTorch from scratch, step by step
Tool for real-time visualization, monitoring and collaborative analysis of AI/ML experiments and live data. Supports Python, PyTorch/Torch, NumPy, TensorFlow/Keras https://visdom.dev
Pytorch domain library for recommendation systems
Native MLX runtime for Laya typed decision models — 7–14 ms short decisions on M3 Max. No text generation, PyTorch, or cloud API.
📄 Awesome OCR multiple programing languages toolkits based on ONNX Runtime, OpenVINO, MNN, PaddlePaddle, TensorRT and PyTorch.
Implementation of rectified flow and some of its followup research / improvements in Pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
Open language modeling toolkit based on PyTorch
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
You like pytorch? You like micrograd? You love tinygrad! ❤️
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
🚀 Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support
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
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.
Graph Neural Network Library for PyTorch
TPU inference for vLLM, with unified JAX and PyTorch support.
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
Datasets, Transforms and Models specific to Computer Vision
Open Lakehouse Format for Multimodal AI. Convert from Parquet in 2 lines of code for 100x faster random access, vector index, and data versioning. Compatible with Pandas, DuckDB, Polars, Pyarrow, and PyTorch with more integrations coming..
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Comprehensive optical design, optimization, and analysis in Python, including GPU-accelerated and differentiable ray tracing via PyTorch.
High-performance implementations of AI kernels for CPUs (x64, AArch64, RISC-V) and Intel GPUs. Powers PyTorch, TensorFlow, OpenVINO, and ONNX Runtime.
Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...
Find why PyTorch training is slow, compare runs, and catch performance regressions in CI.
Capture every activation and gradient of any PyTorch model — forward and backward — with automatic graph visualization, rich metadata, and live interventions. Works on any architecture, including dynamic and recurrent ones.
optimizer & lr scheduler & loss function collections in PyTorch