Jannik Wolff's Projects
BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling
Official PyTorch BIVA implementation (BIVA: A Very Deep Hierarchy of Latent Variables for Generative Modeling)
Basic framework for experiment management in PyTorch.
Code examples in pyTorch and Tensorflow for CS230
A PyTorch re-implementation of "Doubly Reparameterized Gradient Estimators for Monte Carlo Objectives"
Running large language models like ChatGPT/GPT-3/OPT-175B on a single GPU. Up to 100x faster than other offloading systems.
Ladder Variational Autoencoders (LVAE) in PyTorch
Latent Normalizing Flows for Many-to-Many Cross Domain Mappings (ICLR 2020)
A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
Multimodal Mixture-of-Experts VAE
A PyTorch implementation of "Multimodal Generative Models for Scalable Weakly-Supervised Learning" (https://arxiv.org/abs/1802.05335)
The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
Compute FID scores with PyTorch.
A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.
PyTorch deep learning projects made easy.
A latent text-to-image diffusion model
Code for Neural Arithmetic Units (ICLR) and Measuring Arithmetic Extrapolation Performance (SEDL|NeurIPS)
Official PyTorch implementation of StyleGAN3
Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"
Repository for the paper "Very Deep VAEs Generalize Autoregressive Models and Can Outperform Them on Images"