Trending AI reposUpdated daily from GitHub

DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning

Run modern deep learning models in the browser.

D2-Net: A Trainable CNN for Joint Description and Detection of Local Features

Transfer the ControlNet with any basemodel in diffusers🔥

A curated list of awesome work on VAEs, disentanglement, representation learning, and generative models.

Transformer based on a variant of attention that is linear complexity in respect to sequence length

Empower Sequence Labeling with Task-Aware Language Model

Hypernetworks that update LLMs to remember factual information

PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).

Experiments for understanding disentanglement in VAE latent representations

code for our ICCV 2021 paper "DeepCAD: A Deep Generative Network for Computer-Aided Design Models"

A curated list of DATASETS, CODEBASES and PAPERS on Multi-Task Learning (MTL), from Machine Learning perspective.

Play deep learning with CIFAR datasets

Don't memorize ML. Understand it from first principles including Inference, Cuda, Deep Learning, Data Science and Reinforcement Learning

Deep Learning for Coders, 2020, the website

[ICRA 2022] An opensource framework for cooperative detection. Official implementation for OPV2V.

A benchmark dataset for data-driven weather forecasting

Translate - a PyTorch Language Library

A PyTorch implementation of DenseNet.

This is a code repository for pytorch c++ (or libtorch) tutorial.

Generative Adversarial Networks implemented in PyTorch and Tensorflow

Natural Language Toolkit for Indic Languages aims to provide out of the box support for various NLP tasks that an application developer might need

MLGB is a library that includes many models of CTR Prediction & Recommender System by TensorFlow & PyTorch. 「妙计包」是一个包含50+点击率预估和推荐系统深度模型的、通过TensorFlow和PyTorch撰写的库。

Code accompanying the ECCV 2020 paper "Trajectron++: Dynamically-Feasible Trajectory Forecasting With Heterogeneous Data" by Tim Salzmann*, Boris Ivanovic*, Punarjay Chakravarty, and Marco Pavone (* denotes equal contribution).

face recognition algorithms in pytorch framework, including arcface, cosface, sphereface and so on

Collection of papers and resources for data augmentation for NLP.

Refiners (discontinued early 2025) was a microframework on top of PyTorch with first-class citizen APIs for foundation model adaptation.

RL‑Empowered Small‑Scale Competitive Guandan Agent

Official implementation of "MeshDiffusion: Score-based Generative 3D Mesh Modeling" (ICLR 2023 Spotlight)

基于Pytorch和torchtext的自然语言处理深度学习框架。

This repository contains a collection of surveys, datasets, papers, and codes, for predictive uncertainty estimation in deep learning models.

Research papers with annotations, illustrations and explanations

Deep learning with spiking neural networks (SNNs) in PyTorch.

List of Deep Learning Cloud Providers

Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models

A flexible Federated Learning Framework based on PyTorch, simplifying your Federated Learning research.

Robot kinematics implemented in pytorch

A high-fidelity 3D face reconstruction library from monocular RGB image(s)

A very simple CNN project to recognize gestures made in American Sign Language

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.

Descriptive Deep Learning

😎 A list of awesome scene understanding papers.

Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways

:fire::fire: PyTorch implementation of "Progressive growing of GANs (PGGAN)" :fire::fire:

[CVPR 2022] "MonoScene: Monocular 3D Semantic Scene Completion": 3D Semantic Occupancy Prediction from a single image

This is a implementation of the 3D FLAME model in PyTorch

(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"

[ICLR 2024] Controlling Vision-Language Models for Universal Image Restoration. 5th place in the NTIRE 2024 Restore Any Image Model in the Wild Challenge.

A CLI for Kubeflow.

The most easy-to-understand tutorial for using LoRA (Low-Rank Adaptation) within diffusers framework for AI Generation Researchers🔥