Trending AI reposUpdated daily from GitHub

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

This repo contains annotated research papers that I found really good and useful

NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning

Images to inference with no labeling (use foundation models to train supervised models).

《机器翻译：基础与模型》肖桐 朱靖波 著 - Machine Translation: Foundations and Models

List of papers, code and experiments using deep learning for time series forecasting

A Deep Learning Recommender System

基于ChatGLM-6B、ChatGLM2-6B、ChatGLM3-6B模型，进行下游具体任务微调，涉及Freeze、Lora、P-tuning、全参微调等

Learning to Rank in TensorFlow

A toolkit to run Ray applications on Kubernetes

Unsupervised Learning for Image Registration

Learning Chinese Character style with conditional GAN

Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.

MLNLP：本仓库整理人工智能会议（如 ACL、EMNLP、NAACL、COLING、AAAI、IJCAI、ICLR、NeurIPS、ICML 等）中开源代码的论文。

PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"

Real-time end-to-end singing voice conversion system based on DDSP (Differentiable Digital Signal Processing)

Pytorch framework for doing deep learning on point clouds.

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes

Multimodal Unsupervised Image-to-Image Translation

Just draw a bounding box and you can remove the object you want to remove.

Infrastructure for evolutionary AI systems. GPU-accelerated, PyTorch-based optimization with an ecosystem of algorithm libraries, benchmarks, and tools.

BigDL: Distributed TensorFlow, Keras and PyTorch on Apache Spark/Flink & Ray

Deep neural network to extract intelligent information from invoice documents.

Resource scheduling and cluster management for AI

Awesome Knowledge-Distillation. 分类整理的知识蒸馏paper(2014-2021)。

[IEEE TMI Best Paper Award] Official Implementation for UNet++

Deep learning software for colorizing black and white images with a few clicks.

streamline the fine-tuning process for multimodal models: PaliGemma 2, Florence-2, and Qwen2.5-VL

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

FMA: A Dataset For Music Analysis

Build, personalize and control your own LLMs. From data pre-processing to fine-tuning, xTuring provides an easy way to personalize open-source LLMs. Join our discord community: https://discord.gg/TgHXuSJEk6

[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.

Automatically remove the mosaics in images and videos, or add mosaics to them.

Repository for benchmarking graph neural networks (JMLR 2023)

IoT Platform Framework

GeneFace: Generalized and High-Fidelity 3D Talking Face Synthesis; ICLR 2023; Official code

Communicate Freely

Meta Learning / Learning to Learn / One Shot Learning / Few Shot Learning

记录cv算法工程师的成长之路，分享计算机视觉和模型压缩部署技术栈笔记。https://harleyszhang.github.io/cv_note/

Image Deblurring using Generative Adversarial Networks

Awesome Deep Learning papers for industrial Search, Recommendation and Advertisement. They focus on Embedding, Matching, Pre-Ranking, Ranking, Post Ranking, Relevance, LLM and RL. Please cite our paper "Deep Learning to Rank in Industrial Search Engines, Recommender Systems, and Online Advertising - An Overview and New Perspectives" (TOIS 2026).

LeetCode for PyTorch — 65 ML/AI interview problems from real interviews at Google, Meta, Anthropic. Jupyter notebooks, an auto-grader, and an MCP AI tutor.

Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Research, in Pytorch

Collection of generative models in Pytorch version.

Pytorch domain library for recommendation systems

EdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212

TorchCFM: a Conditional Flow Matching library

**Official** 李宏毅 (Hung-yi Lee) 機器學習 Machine Learning 2022 Spring

🐸STT - The deep learning toolkit for Speech-to-Text. Training and deploying STT models has never been so easy.

The PyTorch-based audio source separation toolkit for researchers