hubertrybka / vae-annealingLinks
A simple pytorch implementation for calculating VAE loss components and annealing KLD loss while training VAEs, especially RNN-based
☆20Updated last year
Alternatives and similar repositories for vae-annealing
Users that are interested in vae-annealing are comparing it to the libraries listed below
Sorting:
- Open-source framework for uncertainty and deep learning models in PyTorch☆476Updated this week
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆615Updated last year
- Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)☆1,982Updated last year
- torchview: visualize pytorch models☆1,028Updated 8 months ago
- VICReg official code base☆553Updated 2 years ago
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆727Updated 2 weeks ago
- PyTorch implementation of bayesian neural network [torchbnn]☆553Updated last year
- Lightning-UQ-Box: Uncertainty Quantification for Neural Networks with PyTorch and Lightning☆215Updated last week
- A simple tutorial of Variational AutoEncoders with Pytorch☆432Updated last year
- Learn fast, scalable, and calibrated measures of uncertainty using neural networks!☆508Updated 4 years ago
- VAE for color images☆211Updated 4 years ago
- Repository of the journal club "Diffusion Models and Generative Modeling"☆17Updated last year
- PyTorch implementation of normalizing flow models☆929Updated last year
- Generalized Kraskov Estimator for Normalized Mutual Information☆22Updated last month
- Lightweight, useful implementation of conformal prediction on real data.☆1,011Updated 2 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆499Updated last year
- Example of how to use Weights & Biases on Slurm☆119Updated 3 years ago
- This repository contains a collection of surveys, datasets, papers, and codes, for predictive uncertainty estimation in deep learning mo…☆783Updated 2 months ago
- Code for the paper "Beyond Deep Ensembles: A Large-Scale Evaluation of Bayesian Deep Learning under Distribution Shift"☆22Updated 2 years ago
- Official repository for "Estimating Epistemic and Aleatoric Uncertainty with a Single Model"☆23Updated last year
- TorchCFM: a Conditional Flow Matching library☆2,282Updated 3 months ago
- Gaussian mixture models in PyTorch.☆584Updated 2 years ago
- Normalizing flows in PyTorch☆442Updated 2 months ago
- Codebase for machine learning research in PyTorch.☆14Updated 7 months ago
- A minimal PyTorch implementation of the VQ-VAE model described in "Neural Discrete Representation Learning".☆88Updated 4 years ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆298Updated last week
- Normalizing flows in PyTorch☆991Updated last year
- Quantus is an eXplainable AI toolkit for responsible evaluation of neural network explanations☆639Updated 3 weeks ago
- A simple and extensible library to create Bayesian Neural Network layers on PyTorch.☆978Updated 2 years ago
- Course to learn the basics of self supervised learning☆21Updated last year