Deep Learning

Frameworks for Neural Networks and Deep Learning.

Newest releases

facebookresearch NeuralCompression is a Python repository dedicated to research of neural networks that compress data. The repository includes tools such as JAX-based entropy coders, image compression models, video compression models, and metrics
 

salesforce Align before Fuse: Vision and Language Representation Learning with Momentum Distillation (Salesforce Research)
 

codestella Putting NeRF on a Diet: Semantically Consistent Few-Shot View Synthesis Implementation
 

borisdayma DALL·E Mini - Generate images from a text prompt
 

ermongroup PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations
 

lucas-emery A Gym-like environment for Reinforcement Learning in Rocket League
 

IlyaDobrynin PyTorch optimizations and benchmarking
 

pleaseconnectwifi Learning with Noisy Labels for Robust Point Cloud Segmentation (ICCV2021 Oral)
 

vincentfung13 Code and models for our ICCV 2021 paper "MINE: Towards Continuous Depth MPI with NeRF for Novel View Synthesis"
 

jychoi118 Toward Spatially Unbiased Generative Models (ICCV 2021)
 

Roc-Ng PyTorch implementation of HANet: Hierarchical Alignment Networks for Video-Text Retrieval (ACM MM 2021).
 

yuke93 Path-Restore: Learning Network Path Selection for Image Restoration (IEEE TPAMI, 2021)
 

vision4robotics HiFT: Hierarchical Feature Transformer for Aerial Tracking (ICCV2021)
 

Uason-Chen Official code for "Channel-wise Topology Refinement Graph Convolution for Skeleton-Based Action Recognition"
 

sooftware A PyTorch Implementation of the Luna: Linear Unified Nested Attention
 

hzykent Implementation of ICCV2021(Oral) paper - VMNet: Voxel-Mesh Network for Geodesic-aware 3D Semantic Segmentation
 

AkariAsai This is the official implementation of "One Question Answering Model for Many Languages with Cross-lingual Dense Passage Retrieval".
 

dmsm Code for HodgeNet: Learning Spectral Geometry on Triangle Meshes, in SIGGRAPH 2021.
 

lhc1224 PyTorch implementation of our one-shot affordance detection models. This repository contains PyTorch evaluation code, training code and pretrained models.
 

EleutherAI An open source repository replicating the 2021 paper Knowledge Neurons in Pretrained Transformers by Dai et al., and extending the technique to autoregressive models, as well as MLMs.
 

mindslab-ai UnivNet: A Neural Vocoder with Multi-Resolution Spectrogram Discriminators for High-Fidelity Waveform Generation
 

JiaquanYe This project presents our 2nd place solution for ICDAR 2021 Competition on Scientific Literature Parsing, Task B.
 

mlfoundations An open source implementation of OpenAI's CLIP (Contrastive Language-Image Pre-training).
 

pmj110119 using yolox+deepsort for object-tracking
 

lucidrains Implementation of Multistream Transformers in Pytorch
 

JiaquanYe This project is a re-implementation of MASTER: Multi-Aspect Non-local Network for Scene Text Recognition by MMOCR
 

PruneTruong Dense matching library based on PyTorch
 

lucidrains Implementation of H-Transformer-1D, Transformer using hierarchical Attention for sequence learning with subquadratic costs.
 

patrick-kidger Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.
 

IceTTTb [ICCV'21] PlaneTR: Structure-Guided Transformers for 3D Plane Recovery
 

Guiflayrom YOLNP is a tool that uses YOLOv4 for the extraction of plates in the image, Tesseract and PaddleOCR for OCR. The model for extracting plates and characters has already been completed, leaving only the personalized OCR training wit
 

zhukaii Implementation of the paper "Self-Promoted Prototype Refinement for Few-Shot Class-Incremental Learning"
 

facebookresearch This repository contains a set of tools for working with the Common Objects in 3D (CO3D) dataset.