speechbrain / speechbrain
A PyTorch-based Speech Toolkit
speechbrain / speechbrain
huggingface / accelerate
π A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
bitsandbytes-foundation / bitsandbytes
Accessible large language models via k-bit quantization for PyTorch.
lucidrains / PaLM-rlhf-pytorch
Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM
lyuwenyu / RT-DETR
[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. π₯ π₯ π₯
obss / sahi
Framework agnostic sliced/tiled inference + interactive ui + error analysis plots
apple / coremltools
Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.
awslabs / gluonts
Probabilistic time series modeling in Python
pytorch / executorch
On-device AI across mobile, embedded and edge for PyTorch
sktime / pytorch-forecasting
Time series forecasting with PyTorch
pytorch / ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
onnxsim / onnxsim
Simplify your onnx model
SwanHubX / SwanLab
β‘οΈ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.
ourownstory / neural_prophet
NeuralProphet: A simple forecasting package
torchgeo / torchgeo
TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data
lightly-ai / lightly
A python library for self-supervised learning on images.
NVIDIA / physicsnemo
Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
pytorch / TensorRT
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
thu-pacman / chitu
High-performance inference framework for large language models, focusing on efficiency, flexibility, and availability.
DLR-RM / rl-baselines3-zoo
A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
linto-ai / whisper-timestamped
Multilingual Automatic Speech Recognition with word-level timestamps and confidence
meta-pytorch / torchrec
Pytorch domain library for recommendation systems
AIStream-Peelout / flow-forecast
Deep learning PyTorch library for time series forecasting, classification, and anomaly detection (originally for flood forecasting).
Mukosame / Anime2Sketch
A sketch extractor for anime/illustration.
WenjieDu / PyPOTS
A Python toolkit/library for reality-centric machine/deep learning & data mining on partially-observed time series, with 50+ SOTA neural network models for scientific analysis tasks (imputation, classification, clustering, forecasting, anomaly detection, cleaning) on incomplete industrial irregularly-sampled multivariate TS with NaN missing values
kotaro-kinoshita / yomitoku
YomiTokuγ―AIγζ΄»η¨γγζ₯ζ¬θͺζζΈθ§£ζγ¨γ³γΈγ³γζδΎγγPythonγγγ±γΌγΈγ§γγ Yomitoku is an AI-powered document image analysis package designed specifically for the Japanese language.
Tencent-Hunyuan / AuK
AuK: An Open-Source Foundational Model for Speech Generation and Editing
Topdu / OpenOCR
OpenOCR: An Open-Source Toolkit for General-OCR Research and Applications, integrates a unified training and evaluation benchmark, commercial-grade OCR and Document Parsing systems, and faithful reproductions of the core implementations from a wide range of academic papers.
Lightning-AI / lightning-thunder
PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.
vipshop / cache-dit
A PyTorch-native inference engine with cache, parallelism, quantization and cpu offload for DiTs.
NVIDIA / kvpress
LLM KV cache compression made easy
drprojects / superpoint_transformer
Official PyTorch implementation of Superpoint Transformer [ICCV'23], SuperCluster [3DV'24 Oral], and EZ-SP [ICRA'26]
SakuraMathcraft / LaTeXSnipper
Formula Recognition & Office Editing | Handwriting & PDF to LaTeX/Markdown | Secure API Integrations.
NVIDIA-NeMo / Automodel
π Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support
Rath-Team / OpenRath
An open-source, PyTorch-like runtime for dynamic multi-agent and multi-session workflows.
LibreYOLO / libreyolo
LibreYOLO is a MIT licensed open source computer vision library
NVlabs / SOMA-X
SOMA: Unifying Parametric Human Body Models
albumentations-team / AlbumentationsX
Image augmentation for computer vision. AGPL-3.0-only or commercial licensing.
kyegomez / open-jev
an open-source, from-first-principles reconstruction of the ideas behind TypeSafe AI's Jev, written in pytorch
ZeamoxWang / pdmd
Official Training and Benchmarking Implementation of PDMD: Projected Distribution Matching Distillation for Video Diffusion Models
harrrshall / barunlm-35m
BarunLM-35M: a parameter-efficient 35M base language model trained on 5.7B tokens
Opselon / NexusTradingForexBot
π Enterprise-grade, high-frequency quantitative scalp trading engine for XAUUSD (Gold). Driven by PyTorch (TCN + Self-Attention), Smart Money Concepts (SMC/ICT) , Win32 C++ MT5 IPC bindings, and a real-time FastAPI Canvas Web Dashboard.
intel / gpu-ai-skills
Agent Skills for running, benchmarking, and profiling Hugging Face models on Intel GPUs β PyTorch, vLLM-XPU, and SGLang-XPU
vdev-ctrl / Adaptive-Learnable-Hierarchical-Routing-
This repo is a alternative solution for standard self-attention. It uses tree routing with learnable functions for cheaper self-attention while not compromising accuracy.
Lightning-AI / pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
lucidrains / vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
pyg-team / pytorch_geometric
Graph Neural Network Library for PyTorch
onnx / onnx
Open standard for machine learning interoperability
arc53 / DocsGPT
Private AI platform for agents, assistants and enterprise search. Built-in Agent Builder, Deep research, Document analysis, Multi-model support, and API connectivity for agents.
BlinkDL / RWKV-LM
RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RNN and transformer - great performance, linear time, constant space (no kv-cache), fast training, infinite ctx_len, and free sentence embedding.
flairNLP / flair
A very simple framework for state-of-the-art Natural Language Processing (NLP)
unifyai / ivy
Convert Machine Learning Code Between Frameworks
ludwig-ai / ludwig
Low-code framework for building custom LLMs, neural networks, and other AI models
kornia / kornia
π Geometric Computer Vision Library for Spatial AI
wandb / wandb
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
lucidrains / denoising-diffusion-pytorch
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
ultralytics / yolov3
PyTorch implementation of YOLOv3, YOLOv3-SPP, and YOLOv3-tiny for real-time object detection with training, validation, inference, and multi-format export.
fossasia / visdom
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
OpenMined / PySyft
Perform data science on data that remains in someone else's server
espnet / espnet
End-to-End Speech Processing Toolkit