Ross ROSS = Recommend OSS · open-source software intelligence for agents

domain: deep-learning

2771 products, primary matches first, then adoption-weighted; health v2 shown.

ProductHealth v2StarsMaturity
ZQPei/deep_sort_pytorch
A PyTorch implementation of the Deep SORT multi-object tracking algorithm, pairing YOLOv3/YOLOv5 (or Mask R-CNN) detectors with a CNN re-id…
323012active
deepseek-ai/DualPipe
DualPipe is a Python library implementing a bidirectional pipeline parallelism algorithm that overlaps forward and backward computation wit…
482998active
thu-pacman/chitu
Chitu (赤兔) is a high-performance inference framework for large language models, positioned as a production-grade LLM serving engine. It sup…
842995active
benedekrozemberczki/pytorch_geometric_temporal
PyTorch Geometric Temporal is a temporal (dynamic) extension library for PyTorch Geometric providing spatiotemporal signal processing with …
682992active
MeiGen-AI/MultiTalk
MultiTalk is an audio-driven framework for generating multi-person conversational videos from multi-stream audio, a reference image, and a …
562992active
pytorch/TensorRT
Torch-TensorRT is a compiler library that accelerates PyTorch model inference on NVIDIA GPUs using TensorRT. It supports just-in-time compi…
942986active
Neural Amp Modeler
Neural Amp Modeler (NAM) is a Python project that trains neural network models to emulate guitar amplifiers and exports them as .nam files.…
902986active
sovrasov/flops-counter.pytorch
A Python library (ptflops) that computes the theoretical number of multiply-add operations (FLOPs) and parameter counts for neural network …
412982active
bethgelab/foolbox
Foolbox is a Python library for generating adversarial examples that fool deep neural networks, with state-of-the-art gradient-based and de…
482972active
patrick-kidger/equinox
Equinox is a Python library providing neural networks and scientific computing utilities for JAX, using PyTorch-like class-based syntax whe…
922957stable
pytorch/ao
TorchAO is a PyTorch-native library for model optimization through quantization and sparsity. It supports quantizing weights, gradients, op…
892957active
luminal-ai/luminal
Luminal is a high-performance general-purpose ML inference compiler written in Rust that lowers models to a minimal 15-op dataflow IR and c…
782956active
iscyy/ultralyticsPro
A PyTorch-based collection of improved YOLO-family object detection models (YOLOv5 through YOLOv13, RT-DETR) with pluggable modules for bac…
482954active
karpathy/char-rnn
char-rnn is a Torch/Lua implementation of multi-layer recurrent neural networks (RNN, LSTM, GRU) for character-level language modeling. It …
3212095maintenance
TylerYep/torchinfo
torchinfo is a Python library that provides Keras-style model summaries for PyTorch, showing layer types, input/output shapes, parameter co…
672947active
sunsmarterjie/yolov12
YOLOv12 is a PyTorch implementation of attention-centric real-time object detectors, published at NeurIPS 2025. It provides detection model…
592947active
Biohub/esm
A Python library and model release providing ESMC protein language models, ESMFold2 structure prediction, and the ESM Atlas of 6.8 billion …
742927active
keras-team/keras-tuner
KerasTuner is a hyperparameter optimization framework for Keras models with a define-by-run syntax for configuring search spaces. It ships …
532923active
state-spaces/s4
Official implementations of Structured State Space (S4) sequence models and related variants like S4D, HiPPO, and SaShiMi, built in PyTorch…
322919active
bghira/SimpleTuner
SimpleTuner is a Python fine-tuning toolkit for image, video, and audio diffusion models built on Hugging Face Diffusers. It provides a web…
922912active
allenai/allennlp
AllenNLP is an open-source NLP research library built on PyTorch for developing state-of-the-art deep learning models on linguistic tasks. …
1011886maintenance
Saiyan-World/goku
Goku is a family of flow-based (rectified flow Transformer) foundation models for joint image and video generation, released by HKU and Byt…
232905active
elixir-nx/nx
Nx is an Elixir library providing multi-dimensional arrays (tensors) and numerical definitions with pluggable backends. The monorepo also i…
982901active
decoderesearch/circuit-tracer
A Python library for circuit tracing and attribution graphs in language model interpretability, based on cross-layer MLP transcoders. It co…
792897active
learnables/learn2learn
learn2learn is a PyTorch library for meta-learning research, providing utilities for few-shot task creation, high-level wrappers for algori…
482893active
explosion/thinc
Thinc is a lightweight deep learning library from the makers of spaCy offering a type-checked, functional-programming API for composing neu…
802890active
NVlabs/FoundationStereo
FoundationStereo is NVIDIA's official PyTorch implementation of a foundation model for zero-shot stereo depth estimation, published as a CV…
472874active
werner-duvaud/muzero-general
A commented and documented PyTorch implementation of DeepMind's MuZero reinforcement learning algorithm, designed to be easily adapted to a…
322861active
TMElyralab/MuseV
MuseV is a diffusion-based framework for generating high-fidelity virtual human videos of infinite length using a Visual Conditioned Parall…
252846active
OpenGVLab/InternImage
InternImage is a large-scale CNN-based vision foundation model that uses deformable convolutions as its core operator, released with pretra…
282841stable
adapter-hub/adapters
Adapters is a Python add-on library for HuggingFace Transformers that integrates 10+ parameter-efficient fine-tuning methods (bottleneck ad…
802826active
pytorch/xla
PyTorch/XLA is a Python package that connects the PyTorch deep learning framework to XLA devices such as Google Cloud TPUs via the XLA deep…
692803active
KellerJordan/Muon
Muon is a PyTorch optimizer for the hidden layers of neural networks, based on orthogonalized momentum updates via Newton-Schulz iteration.…
592801active
huggingface/nanotron
Nanotron is a minimalistic Python library from Hugging Face for pretraining large language models with 3D parallelism (data, tensor, and pi…
562800active
lucidrains/DALLE2-pytorch
A PyTorch implementation of OpenAI's DALL-E 2 text-to-image synthesis model, focusing on the diffusion prior network that predicts image em…
2311306maintenance
Mega4alik/ollm
oLLM is a lightweight Python library for large-context LLM inference built on Hugging Face Transformers and PyTorch. It offloads weights an…
602788active
huggingface/setfit
SetFit is a Python library for efficient, prompt-free few-shot fine-tuning of Sentence Transformers for text classification. It achieves hi…
642784active
FasterDecoding/Medusa
Medusa is a framework that accelerates LLM text generation by adding multiple decoding heads to an existing model, avoiding the need for a …
182770active
ideogram-oss/ideogram4
Ideogram 4 is an open-weight text-to-image foundation model trained from scratch, with inference code and weights released in Python. It fe…
542766active
NVlabs/stylegan2
The official TensorFlow implementation of StyleGAN2, NVIDIA's improved style-based generative adversarial network for high-quality uncondit…
3211184maintenance
AutoArk/GPA
GPA (General Purpose Audio) is a unified autoregressive audio-language model that performs text-to-speech, automatic speech recognition, an…
542762active
NVIDIA/FastPhotoStyle
FastPhotoStyle is NVIDIA's official PyTorch implementation of the ECCV 2018 paper 'A Closed-form Solution to Photorealistic Image Stylizati…
2311177maintenance
apple/turicreate
Turi Create is a Python library from Apple that simplifies building custom machine learning models for tasks like image classification, obj…
1011159maintenance
voxelmorph/voxelmorph
VoxelMorph is a Python library for learning-based image registration and alignment, using unsupervised deep learning to model deformations …
762748active
ModelTC/LightX2V
LightX2V is a lightweight, high-performance inference framework for image and video generation, supporting tasks like text-to-video, image-…
642733active
artidoro/qlora
QLoRA is the official implementation of the QLoRA paper, an efficient finetuning approach that backpropagates through a frozen 4-bit quanti…
2910998maintenance
prophesier/diff-svc
Diff-SVC is a deep learning project that performs singing voice conversion using diffusion models, transforming input singing audio into a …
622717active
lengstrom/fast-style-transfer
A TensorFlow implementation of fast neural style transfer that applies the style of famous paintings to photos and videos in real time. It …
3210962maintenance
bmild/nerf
The official TensorFlow implementation of NeRF (Neural Radiance Fields), the ECCV 2020 paper representing scenes as neural radiance fields …
3910927maintenance
intel/neural-compressor
Intel Neural Compressor is an open-source Python library providing state-of-the-art low-bit quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4),…
922704active
yuweihao/MambaOut
MambaOut is a PyTorch implementation of Gated CNN models from the CVPR 2025 paper 'MambaOut: Do We Really Need Mamba for Vision?', which qu…
192704stable
TMElyralab/MusePose
MusePose is a diffusion-based, pose-guided image-to-video generation framework for creating virtual human videos, where a character in a re…
282701active
huggingface/text-generation-inference
Text Generation Inference (TGI) is a Rust, Python and gRPC toolkit for deploying and serving large language models with high performance, p…
1010889maintenance
xdit-project/xDiT
xDiT is a scalable inference engine for Diffusion Transformers (DiTs) that enables parallel deployment across multiple GPUs and machines. I…
772699active
SkyworkAI/SkyReels-V1
SkyReels V1 is an open-source human-centric video foundation model with Text-to-Video and Image-to-Video variants, fine-tuned from HunyuanV…
252696active
roboflow/maestro
maestro is a Python library from Roboflow that streamlines fine-tuning of multimodal vision-language models such as Florence-2, PaliGemma 2…
622694active
baaivision/EVA
EVA is a family of large-scale vision foundation models from BAAI, including masked image models (EVA-01/02) and scaled CLIP models (EVA-CL…
232691active
openai/DALL-E
The official PyTorch package for the discrete VAE (dVAE) component of OpenAI's DALL·E model. It does not include the transformer that gener…
1010834maintenance
qualcomm/aimet
AIMET (AI Model Efficiency Toolkit) is a Python library from Qualcomm providing advanced quantization and compression techniques for traine…
992688active
bytedance/InfiniteYou
InfiniteYou (InfU) is a research framework from ByteDance for identity-preserved text-to-image generation built on Diffusion Transformers l…
372685active
yuqinie98/PatchTST
Official PyTorch implementation of PatchTST, an ICLR 2023 Transformer model for long-term time series forecasting based on patching and cha…
322685stable
MrGiovanni/UNetPlusPlus
Official implementation of UNet++, a nested U-Net architecture for medical image segmentation, in both Keras and PyTorch. It redesigns skip…
772679stable
HiLab-git/SSL4MIS
A benchmark and code collection of semi-supervised learning methods for medical image segmentation, re-implementing approaches like Mean Te…
442676active
stochasticai/xTuring
xTuring is a Python library for fine-tuning, evaluating, and running open-source large language models such as LLaMA, GPT-J, GPT-2, Qwen, a…
522674active
ZHZisZZ/dllm
dLLM is a Python library that unifies training, inference, and evaluation of diffusion language models such as LLaDA and Dream. It builds o…
592672active
IceClear/StableSR
StableSR is a Python research library that leverages pre-trained Stable Diffusion priors for real-world blind image super-resolution. It pr…
212668stable
google-deepmind/mctx
Mctx is a JAX-native Python library implementing Monte Carlo tree search algorithms such as AlphaZero, MuZero, and Gumbel MuZero. It suppor…
832654active
phillipi/pix2pix
The original Torch (Lua) implementation of pix2pix, a conditional GAN for image-to-image translation tasks such as synthesizing photos from…
3210652maintenance
Tencent/MimicMotion
MimicMotion is a diffusion-based framework from Tencent for generating high-quality human motion videos guided by pose sequences, featuring…
472647active
black-forest-labs/flux2
Official inference repository for Black Forest Labs' FLUX.2 family of open-weight image generation and editing models. It provides minimal …
482642active
ultralytics/yolov3
Ultralytics' PyTorch implementation of YOLOv3, YOLOv3-SPP, and YOLOv3-tiny for real-time object detection. It provides training, validation…
6710596maintenance
anliyuan/Ultralight-Digital-Human
An ultralight talking-head (digital human) model that animates a person's face from audio input and runs in real time on mobile devices. It…
642627active
lucidrains/audiolm-pytorch
A PyTorch implementation of AudioLM, Google Research's language modeling approach to audio generation, including a MIT-licensed SoundStream…
342627active
swz30/Restormer
Restormer is an efficient Transformer architecture for high-resolution image restoration, published as a CVPR 2022 Oral paper. It provides …
442625stable
open-gigaai/giga-brain-0
GigaBrain-0/0.7 is an open-source vision-language-action (VLA) model family for generalist embodied agents, powered by world models and a t…
622611active
crowsonkb/k-diffusion
A PyTorch library implementing Karras et al. (2022) diffusion models with enhancements like improved sampling algorithms and transformer-ba…
532600active
meta-pytorch/torchrec
TorchRec is a PyTorch domain library for building recommendation systems at scale. It provides distributed sharding of large embedding tabl…
902599active
kijai/ComfyUI-HunyuanVideoWrapper
A set of custom ComfyUI nodes that wrap Tencent's HunyuanVideo text-to-video and image-to-video diffusion model for use inside ComfyUI work…
382597active
dreamzero0/dreamzero
DreamZero is NVIDIA's World Action Model (WAM) that jointly predicts future video and actions from a pretrained video diffusion backbone, e…
502593active
facebookresearch/demucs
Demucs is a state-of-the-art music source separation model from Meta AI that splits songs into stems like drums, bass, and vocals using a h…
1010359maintenance
median-research-group/LibMTL
LibMTL is an open-source PyTorch library for Multi-Task Learning (MTL). It provides implementations of many MTL architectures and gradient-…
422586active
asteroid-team/asteroid
Asteroid is a PyTorch-based audio source separation toolkit for researchers, providing modular building blocks (filterbanks, encoders, mask…
602584active
atong01/conditional-flow-matching
TorchCFM is a PyTorch library implementing Conditional Flow Matching (CFM), a simulation-free training objective for continuous normalizing…
742571active
Tencent-Hunyuan/HY-World-2.0
HY-World 2.0 is Tencent Hunyuan's open-source multi-modal world model framework that reconstructs, generates, and simulates 3D worlds from …
582571active
Memento-Teams/Memento
Memento is a Python framework for building LLM agents that continually improve from experience via memory-based case-based reasoning, witho…
392568active
NVlabs/LongLive
LongLive is an NVIDIA research framework providing parallel training and inference infrastructure for real-time long video generation, usin…
602563active
advimman/lama
LaMa is a PyTorch-based image inpainting model that fills large missing regions in images using fast Fourier convolutions, generalizing wel…
3410217maintenance
KomputeProject/kompute
Kompute is a general-purpose GPU compute framework built on Vulkan that works across vendor GPUs (AMD, NVIDIA, Qualcomm, etc.) with both C+…
662558active
vita-epfl/Stable-Video-Infinity
Stable Video Infinity (SVI) is a research codebase for infinite-length video generation using video diffusion transformers with an error-re…
552556active
jolibrain/deepdetect
DeepDetect is an open-source deep learning runtime, CLI, and REST server written in C++ for training and inference across images, text, tab…
952551active
mozilla/TTS
A deep learning library for advanced text-to-speech generation, built on PyTorch with models like Tacotron2, Glow-TTS, and various vocoders…
2310167maintenance
Tencent-Hunyuan/HY-Motion-1.0
HY-Motion 1.0 is a series of text-to-3D human motion generation models from Tencent Hunyuan, based on Diffusion Transformer and Flow Matchi…
582519active
learning-at-home/hivemind
Hivemind is a PyTorch library for decentralized deep learning across the Internet, enabling training of large models on hundreds of volunte…
592515active
rpautrat/SuperPoint
A TensorFlow (with PyTorch conversion) implementation of the SuperPoint self-supervised interest point detector and descriptor network. It …
412511stable
SafeAILab/EAGLE
EAGLE is the official implementation of the EAGLE family of speculative decoding algorithms (EAGLE-1/2/3) for accelerating LLM text generat…
572510active
luanfujun/deep-photo-styletransfer
Reference implementation of the CVPR 2017 paper 'Deep Photo Style Transfer', performing photorealistic image style transfer using Torch wit…
329989maintenance
KohakuBlueleaf/LyCORIS
LyCORIS is a Python library implementing parameter-efficient fine-tuning algorithms (LoRA/LoCon, LoHa, LoKr, IA3, DyLoRA, and more) for Sta…
732508active
LTH14/JiT
A PyTorch/GPU re-implementation of JiT (Just image Transformer), a minimalist pixel-space diffusion model for high-resolution image generat…
422507active
bilibili/Index-anisora
Index-AniSora is Bilibili's open-source anime video generation model, capable of creating video shots in diverse anime styles from images, …
622505active
google-deepmind/sonnet
Sonnet is a Python library built on TensorFlow 2 that provides simple, composable abstractions (snt.Module) for constructing neural network…
649961maintenance

← prev page 6 / 28 next →