vgsatorras

few-shot-gnn

egnn

en_flows

hybrid-inference

probai

Deep Generative Models: Diffusion Models for Molecule Generation

DCRNN_PyTorch

Diffusion Convolutional Recurrent Neural Network Implementation in PyTorch

pytorch-rl-1

Deep Reinforcement Learning with pytorch & visdom

graph-generation

GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Models

baselines

OpenAI Baselines: high-quality implementations of reinforcement learning algorithms

convolution_exponential_and_sylvester

cv19index

COVID-19 Vulnerability Index

DCRNN

Implementation of Diffusion Convolutional Recurrent Neural Network in Tensorflow

gitignore

A collection of useful .gitignore templates

gluon-ts

Probabilistic time series modeling in Python

Graph-WaveNet

graph wavenet

GTS

Discrete Graph Structure Learning for Forecasting Multiple Time Series, ICLR 2021.

mace

MACE implementation in PyTorch

Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

mujoco-py

MuJoCo is a physics engine for detailed, efficient rigid body simulations with contacts. mujoco-py allows using MuJoCo from Python 3.

mypy

Optional static typing for Python 3 and 2 (PEP 484)

NRI

Neural relational inference for interacting systems - pytorch

opencv_extra

OpenCV extra data

pygcn

Graph Convolutional Networks in PyTorch

PyTorch-ActorCriticRL

PyTorch implementation of DDPG algorithm for continuous action reinforcement learning problem.

random

roboschool

Open-source software for robot simulation, integrated with OpenAI Gym.

se3-transformer-public

code for the SE3 Transformers paper: https://arxiv.org/abs/2006.10503

siren-pytorch

Pytorch implementation of SIREN - Implicit Neural Representations with Periodic Activation Function

tutorials

PyTorch tutorials.

vision

Datasets, Transforms and Models specific to Computer Vision