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
open-mmlab/mmflow
MMFlow is an open-source optical flow toolbox and benchmark built on PyTorch as part of the OpenMMLab project. It provides a unified, modul…
231062maintenance
zhaoweicai/cascade-rcnn
A C++/Caffe implementation of Cascade R-CNN and other popular two-stage object detection frameworks such as Faster R-CNN, R-FCN, and FPN. I…
321061maintenance
OFA-Sys/ONE-PEACE
ONE-PEACE is a general multimodal representation model that jointly encodes vision, audio, and language modalities without initializing fro…
291060maintenance
philipperemy/keract
Keract is a Python library that extracts layer activations (outputs) and gradients from TensorFlow/Keras models as NumPy arrays, with helpe…
301059maintenance
chenyuntc/PyTorchText
A PyTorch implementation of multiple text classification models (TextCNN, TextRNN/LSTM, RCNN, FastText, inception CNN) that won 1st place i…
321057maintenance
HRNet/HRNet-Image-Classification
Official PyTorch implementation and training code for HRNet (High-Resolution Network) image classification models on ImageNet. It provides …
231056maintenance
RexYing/gnn-model-explainer
Official implementation of GNNExplainer, a NeurIPS 2019 method for explaining predictions of graph neural networks by learning soft masks o…
321055maintenance
keras-team/keras-cv
KerasCV is a library of modular computer vision components built on Keras 3 that work natively with TensorFlow, JAX, or PyTorch. It provide…
101055maintenance
piergiaj/pytorch-i3d
A PyTorch port of DeepMind's I3D (Inflated 3D ConvNet) models pretrained on the Kinetics dataset for video action recognition. It includes …
321054maintenance
huggingface/optimum-quanto
Optimum Quanto is a PyTorch quantization backend for Hugging Face Optimum that quantizes model weights (int2/int4/int8/float8) and activati…
701053maintenance
raminmh/CfC
Reference implementations of Closed-form Continuous-time (CfC) neural networks, a fast closed-form approximation of liquid time-constant ne…
231053maintenance
microsoft/Oscar
Oscar is Microsoft's research code for object-semantics aligned cross-modal pre-training of vision-language models, with VinVL providing im…
101053maintenance
yangxudong/deeplearning
A collection of deep learning model training, evaluation, and prediction code implemented with TensorFlow's high-level Estimator API, with …
321049maintenance
damo-cv/TransReID
Official PyTorch implementation of TransReID, an ICCV 2021 paper applying vision transformers to object re-identification. It provides trai…
321048maintenance
microsoft/SimMIM
Official PyTorch implementation of SimMIM, a simple framework for masked image modeling (self-supervised visual pre-training) from Microsof…
321048maintenance
thunlp/OpenDelta
OpenDelta is a Python library for parameter-efficient tuning (delta tuning) of pretrained language models, letting users attach small train…
231046maintenance
4uiiurz1/pytorch-nested-unet
A PyTorch implementation of the UNet++ (Nested U-Net) architecture for image segmentation, based on the paper 'UNet++: A Nested U-Net Archi…
321045maintenance
YuwenXiong/py-R-FCN
A Python implementation of R-FCN (Region-based Fully Convolutional Networks) for object detection, modified from the official MATLAB code a…
321043maintenance
facebookresearch/FixRes
FixRes is a PyTorch implementation of the NeurIPS 2019 paper 'Fixing the train-test resolution discrepancy', providing training and fine-tu…
101043maintenance
asingh33/CNNGestureRecognizer
A desktop application that recognizes hand gestures from webcam video using a convolutional neural network built with Keras, TensorFlow/The…
601041maintenance
SysCV/sam-pt
SAM-PT extends the Segment Anything Model to zero-shot video segmentation by combining SAM with sparse point-based tracking (PIPS, CoTracke…
291041maintenance
keijiro/Pix2Pix
A Unity library that runs pix2pix image-to-image translation neural networks in real time using compute shaders. It includes its own infere…
231041maintenance
descriptinc/melgan-neurips
Official PyTorch implementation of MelGAN, a GAN-based non-autoregressive vocoder that inverts mel-spectrograms into raw audio waveforms fo…
321039maintenance
huggingface/pytorch-pretrained-BigGAN
A PyTorch reimplementation of DeepMind's BigGAN generator with pretrained weights at 128, 256, and 512 pixel resolutions, plus scripts to c…
231039maintenance
google-research/morph-net
MorphNet is a TensorFlow library for resource-constrained learning of deep network structure. It adds regularizers during training that ind…
731038maintenance
JIA-Lab-research/SNR-Aware-Low-Light-Enhance
Official PyTorch implementation of the CVPR 2022 paper 'SNR-aware Low-Light Image Enhancement'. It combines SNR-aware transformers and conv…
321037maintenance
CalciferZh/minimal-hand
A Python library for real-time 3D hand motion capture from a single RGB camera, running at over 100fps. It combines a deep network (DetNet)…
231037maintenance
sicara/tf-explain
tf-explain is a Python library implementing interpretability methods (Grad CAM, occlusion sensitivity, vanilla gradients, etc.) for tf.kera…
231037maintenance
google-research/deeplab2
DeepLab2 is a TensorFlow library from Google Research providing a unified, state-of-the-art codebase for dense pixel labeling tasks such as…
101037maintenance
Kismuz/btgym
BTGym is a Python library that wraps the Backtrader algorithmic trading engine in an OpenAI Gym environment API, enabling event-driven back…
321034maintenance
pesser/stable-diffusion
The development repository for Stable Diffusion and Latent Diffusion Models, containing research code, training scripts, and pretrained mod…
321033maintenance
goberoi/faceit
A Python script that simplifies swapping faces in videos using the deepfakes/faceswap library, with training data sourced from YouTube vide…
321032maintenance
ClementPinard/SfmLearner-Pytorch
A PyTorch reimplementation of SfMLearner (Zhou et al., CVPR 2017), which jointly learns monocular depth estimation and camera ego-motion fr…
671031maintenance
xingyizhou/ExtremeNet
Official PyTorch implementation of ExtremeNet, a CVPR 2019 bottom-up object detection method that detects four extreme points and one cente…
321031maintenance
kakaobrain/rq-vae-transformer
The official PyTorch implementation of 'Autoregressive Image Generation using Residual Quantization' (CVPR 2022), implementing RQ-VAE and R…
321030maintenance
qubvel/ttach
TTAch is a Python library for image test time augmentation (TTA) with PyTorch. It wraps existing models to apply augmentations like flips, …
231030maintenance
vincentherrmann/pytorch-wavenet
A PyTorch implementation of the WaveNet neural audio generation architecture with fast generation support. It includes dataset creation fro…
321029maintenance
google-deepmind/scalable_agent
A TensorFlow implementation of IMPALA (Importance Weighted Actor-Learner Architectures) for scalable distributed deep reinforcement learnin…
321028maintenance
graphdeeplearning/graphtransformer
Reference implementation of the Graph Transformer architecture from the paper 'A Generalization of Transformer Networks to Graphs' (DLG-AAA…
321028maintenance
yuval-alaluf/hyperstyle
Official PyTorch implementation of HyperStyle (CVPR 2022), a hypernetwork that inverts real images into editable regions of StyleGAN's late…
321027maintenance
lmb-freiburg/flownet2
A Caffe fork implementing FlowNet 2.0, a deep CNN for optical flow estimation from image pairs, released with the CVPR 2017 paper. It inclu…
321024maintenance
cxxnet
ps-lite is a lightweight, efficient C++ implementation of the parameter server framework for distributed machine learning. It exposes simpl…
101024maintenance
awslabs/multi-model-server
Multi Model Server (MMS) is a tool for serving deep learning model inference over HTTP endpoints, supporting models from any ML/DL framewor…
101024maintenance
turtlesoupy/this-word-does-not-exist
A project that trains a GPT-2 variant to invent fake English words with generated definitions and example sentences, powering the thiswordd…
721023maintenance
sniklaus/sepconv-slomo
A reference PyTorch implementation of Video Frame Interpolation via Adaptive Separable Convolution, which generates intermediate frames bet…
431021maintenance
Gumpest/YOLOv5-Multibackbone-Compression
A YOLOv5-based toolbox for swapping in lightweight or high-accuracy backbones (TPH-YOLOv5, GhostNet, ShuffleNetV2, MobileNetV3-Small, Effic…
321020maintenance
tensorflow/similarity
TensorFlow Similarity is a Python library built on TensorFlow for training and serving similarity and metric learning models, including sel…
101020maintenance
bayesiains/nflows
nflows is a comprehensive PyTorch library of normalizing flows for density estimation and generative modeling. It provides invertible trans…
231019maintenance
brain-research/self-attention-gan
A TensorFlow implementation of Self-Attention GANs for reproducing results from the paper 'Self-Attention Generative Adversarial Networks' …
101019maintenance
EvgenyKashin/stylegan2-distillation
A research implementation of the ECCV 2020 paper 'StyleGAN2 Distillation for Feed-forward Image Manipulation', distilling StyleGAN2 latent-…
321018maintenance
facebookresearch/barlowtwins
A PyTorch implementation of Barlow Twins, a self-supervised learning method that trains visual representations by reducing redundancy betwe…
101017maintenance
AMLab-Amsterdam/AttentionDeepMIL
A PyTorch implementation of the paper 'Attention-based Deep Multiple Instance Learning' (Ilse, Tomczak & Welling, 2018). It provides code t…
321012maintenance
Oldpan/Pytorch-Memory-Utils
A small set of Python scripts for tracking GPU memory usage in PyTorch, including per-line tensor allocation reports and model parameter/in…
321012maintenance
snap-research/NeROIC
Official PyTorch implementation of NeROIC, a neural method for capturing 3D object geometry and material from online image collections and …
321012maintenance
wpeebles/gangealing
Official PyTorch implementation of GANgealing, a CVPR 2022 method that trains a Spatial Transformer to densely align images using GAN-gener…
321012maintenance
tensorflow/neural-structured-learning
Neural Structured Learning (NSL) is a TensorFlow framework for training neural networks with structured signals, either explicit graphs or …
651010maintenance
allegro/allRank
allRank is a PyTorch-based framework for training neural learning-to-rank (LTR) models. It provides pointwise, pairwise, and listwise loss …
321009maintenance
zhanghang1989/PyTorch-Multi-Style-Transfer
A PyTorch implementation of MSG-Net and Gatys et al. neural style transfer for applying artistic styles to images in real time. It includes…
231009maintenance
xiaoyufenfei/Efficient-Segmentation-Networks
A PyTorch reference implementation collection of lightweight, real-time semantic segmentation models such as ENet, ERFNet, LEDNet, Fast-SCN…
321008maintenance
PeterWang512/CNNDetection
A PyTorch research codebase with pretrained models for detecting CNN-generated (GAN/synthetic) images, from the CVPR 2020 paper 'CNN-genera…
321005maintenance
kevinzakka/spatial-transformer-network
A TensorFlow implementation of Spatial Transformer Networks, a differentiable module that can be inserted into ConvNet architectures to add…
321005maintenance
mileyan/pseudo_lidar
Research code implementing Pseudo-LiDAR, a CVPR 2019 method that converts image-based depth maps into pseudo-LiDAR point clouds for 3D obje…
321005maintenance
bubbliiiing/yolov8-pytorch
A PyTorch implementation of the YOLOv8 object detection model with training, prediction, and evaluation scripts. It supports training on cu…
211005maintenance
alex04072000/ObstructionRemoval
The official TensorFlow implementation of the CVPR 2020 paper 'Learning to See Through Obstructions', which removes obstructions like windo…
321004maintenance
johannakarras/DreamPose
Official PyTorch implementation of DreamPose, a Stable Diffusion-based model that synthesizes animated fashion videos from a single image a…
301004maintenance
NATSpeech/NATSpeech
A PyTorch framework for non-autoregressive text-to-speech (NAR-TTS), containing official implementations of PortaSpeech (NeurIPS 2021) and …
231004maintenance
LeeSureman/Flat-Lattice-Transformer
Reference implementation of the ACL 2020 paper FLAT: Chinese NER Using Flat-Lattice Transformer, built on PyTorch and FastNLP. It trains fl…
321003maintenance
shaoshengsong/DeepSORT
A C++ implementation of multi-object tracking (MOT) combining YOLOv5 object detection with DeepSORT and ByteTrack trackers. It uses ONNX Ru…
321003maintenance
microsoft/nnfusion
NNFusion is a flexible and efficient deep neural network (DNN) compiler that generates high-performance executables from model descriptions…
231002maintenance
google-research/magvit
Official JAX implementation of MAGVIT, a masked generative video transformer from a CVPR 2023 paper by Google Research and CMU. It provides…
101000maintenance
openai/grok
Research code accompanying the paper 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' from OpenAI. It provides t…
104263experimental
0hq/WebGPT
WebGPT is a vanilla JavaScript and HTML implementation of GPT transformer inference running in the browser via WebGPU, in under ~1500 lines…
303792experimental
MoonshotAI/Attention-Residuals
Official implementation of Attention Residuals (AttnRes), a drop-in replacement for standard residual connections in Transformers that lets…
473487experimental
lucidrains/musiclm-pytorch
A PyTorch library implementing MusicLM, Google's text-to-music generation model, by combining text-conditioned AudioLM with MuLan, a text-a…
213293experimental
enhuiz/vall-e
An unofficial PyTorch implementation of the VALL-E text-to-speech audio language model, built on the EnCodec tokenizer. It provides trainin…
312976experimental
guoqincode/Open-AnimateAnyone
An unofficial PyTorch implementation of Animate Anyone, a diffusion-based method that animates a static character image using pose sequence…
262923experimental
Everlyn-Labs/Everlyn-1
Everlyn-1 is an open autoregressive foundational video AI model from Everlyn Labs, accompanied by research on video compression/tokenizatio…
222892experimental
test-time-training/ttt-video-dit
Official PyTorch implementation of 'One-Minute Video Generation with Test-Time Training', which finetunes the CogVideoX 5B diffusion transf…
502448experimental
dvmazur/mixtral-offloading
A Python library enabling efficient inference of Mixtral-8x7B mixture-of-experts language models on limited hardware like Google Colab or c…
262332experimental
google/tangent
Tangent is a Python library for automatic differentiation that works by source-to-source transformation: it reads your Python function's so…
102328experimental
JiauZhang/DragGAN
A Python implementation of DragGAN, a research method for interactively manipulating generated images by dragging points on the generative …
292128experimental
etched-ai/open-oasis
Inference code and model weights for Oasis 500M, an interactive world model from Decart and Etched that generates gameplay video autoregres…
222123experimental
lucidrains/make-a-video-pytorch
A PyTorch library implementing Make-A-Video, Meta AI's text-to-video generation approach, built around pseudo-3d (axial) convolutions and s…
231986experimental
lucidrains/gigagan-pytorch
A PyTorch implementation of GigaGAN, Adobe's state-of-the-art generative adversarial network for text-to-image and unconditional image synt…
211942experimental
chelsea0x3b/dfdx
A Rust deep learning library providing shape-checked tensors and neural network building blocks with compile-time verification of tensor op…
231932experimental
matthewearl/deep-anpr
An experimental Python project that uses neural networks (TensorFlow) to build an automatic number plate recognition (ANPR) system. It incl…
321856experimental
rentruewang/aioway
Aioway is an optimizing compiler for deep learning algorithms that treats ML models as instructions and builds declarative pipelines using …
771824experimental
deepseek-ai/DeepSeek-V3.2-Exp
DeepSeek-V3.2-Exp is an experimental open-weight large language model release that introduces DeepSeek Sparse Attention for more efficient …
401640experimental
Anything-of-anything/Anything-3D
Anything-3D is a Python research project that combines Meta's Segment Anything model with a series of 3D models (3DFuse, Zero 1-to-3, NeRF,…
301633experimental
EvolvingLMMs-Lab/open-r1-multimodal
A fork of Hugging Face's open-r1 that extends the R1 GRPO reinforcement learning training paradigm to multimodal (vision-language) models l…
231603experimental
lichao-sun/Mora
Mora is an open-source multi-agent framework for generalist video generation, using multiple collaborative visual AI agents to replicate ca…
251592experimental
lyuchenyang/Macaw-LLM
Macaw-LLM is a multi-modal language modeling framework that integrates image, video, audio, and text data, built on CLIP, Whisper, and LLaM…
291591experimental
AnswerDotAI/fsdp_qlora
A training script/library from Answer.AI that combines QLoRA (quantized LoRA) with PyTorch FSDP to fine-tune large language models like Lla…
261550experimental
KUR-creative/SickZil-Machine
SickZil-Machine is a desktop application that automates text removal from manga and comic pages during the scanlation (translation) process…
231524experimental
test-time-training/ttt-lm-pytorch
Official PyTorch implementation of Test-Time Training (TTT) layers, a sequence modeling layer with linear complexity whose hidden state is …
231388experimental
lucidrains/video-diffusion-pytorch
A Pytorch implementation of the Video Diffusion Models paper by Jonathan Ho, extending denoising diffusion probabilistic models (DDPMs) to …
231383experimental
lizhihao6/Sparc3D
Sparc3D is the official implementation of a research framework for high-resolution 3D shape modeling, combining a sparse deformable marchin…
311353experimental
lucidrains/naturalspeech2-pytorch
A PyTorch implementation of NaturalSpeech 2, a zero-shot text-to-speech and singing synthesizer that combines a neural audio codec with a l…
201333experimental
google/style-aligned
Official research code for 'Style Aligned Image Generation via Shared Attention', implementing style-consistent image generation with diffu…
101315experimental
Jamie-Stirling/RetNet
A minimal, pure PyTorch implementation of the Retentive Network (RetNet) architecture proposed as a successor to Transformers for large lan…
281209experimental

← prev page 23 / 28 next →