Stars
Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"
[CVPR 2020 Workshop] A PyTorch GAN library that reproduces research results for popular GANs.
A Prior of a Googol Gaussians: a Tensor Ring Induced Prior for Generative Models
Generative Tensorial Reinforcement Learning (GENTRL) model
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
PlaidML is a framework for making deep learning work everywhere.
lagom: A PyTorch infrastructure for rapid prototyping of reinforcement learning algorithms.
Junction Tree Variational Autoencoder for Molecular Graph Generation (ICML 2018)
Molecular Sets (MOSES): A Benchmarking Platform for Molecular Generation Models