szagoruyko

YaLM-100B

Pretrained language model with 100B parameters

Kandinsky-2

Kandinsky 2 — multilingual text2image latent diffusion model

wahooMapsCreator

Create maps for Wahoo device based on latest OSM maps

detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

mdetr

detr-tensorflow

Tensorflow implementation of DETR : Object Detection with Transformers

a-shell

A terminal for iOS, with multiple windows

swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

detr

End-to-End Object Detection with Transformers

submitit

Python 3.8+ toolbox for submitting jobs to Slurm

TorchCraftAI

A platform that lets you build agents to learn to play StarCraft: Brood War.

tvm

TVM integration into PyTorch

PyRep

A toolkit for robot learning research.

rearrangement-planning

Vision code for "Monte-Carlo Tree Search for Efficient Visually Guided Rearrangement Planning", RA-L 2020.

PracticalDeepStereo_NIPS2018

Code for "Practical Deep Stereo (PDS): Toward applications-friendly deep stereo matching", NeurIPS 2018.

PlotNeuralNet

Latex code for making neural networks diagrams

visual-pushing-grasping

Train robotic agents to learn to plan pushing and grasping actions for manipulation with deep reinforcement learning.

tetris_mcts

MCTS project for Tetris

pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!

maskrcnn-benchmark

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

quantile-regression-dqn-pytorch

A short and easy implementation of Quantile Regression DQN | Distributional Reinforcement Learning

ppocma

kornia

Geometric Computer Vision Library for Spatial AI

lagom

lagom: A PyTorch infrastructure for rapid prototyping of reinforcement learning algorithms.

bodynet

BodyNet: Volumetric Inference of 3D Human Body Shapes, ECCV 2018

awesome-knowledge-distillation

Awesome Knowledge Distillation

1-bit-per-weight

Training wide residual networks for deployment using a single bit for each weight

jupyter_tensorboard

Start Tensorboard in Jupyter Notebook

distiller

Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller

AutoAugment

Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow