Stars
Pretrained language model with 100B parameters
Kandinsky 2 — multilingual text2image latent diffusion model
Create maps for Wahoo device based on latest OSM maps
detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.
Tensorflow implementation of DETR : Object Detection with Transformers
PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882
End-to-End Object Detection with Transformers
Python 3.8+ toolbox for submitting jobs to Slurm
A platform that lets you build agents to learn to play StarCraft: Brood War.
Vision code for "Monte-Carlo Tree Search for Efficient Visually Guided Rearrangement Planning", RA-L 2020.
Code for "Practical Deep Stereo (PDS): Toward applications-friendly deep stereo matching", NeurIPS 2018.
Latex code for making neural networks diagrams
Train robotic agents to learn to plan pushing and grasping actions for manipulation with deep reinforcement learning.
Efficient PyTorch Hessian eigendecomposition tools!
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
A short and easy implementation of Quantile Regression DQN | Distributional Reinforcement Learning
Geometric Computer Vision Library for Spatial AI
lagom: A PyTorch infrastructure for rapid prototyping of reinforcement learning algorithms.
BodyNet: Volumetric Inference of 3D Human Body Shapes, ECCV 2018
Awesome Knowledge Distillation
Training wide residual networks for deployment using a single bit for each weight
Start Tensorboard in Jupyter Notebook
Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller
Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow