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☆470Updated last week
- Lightning-UQ-Box: Uncertainty Quantification for Neural Networks with PyTorch and Lightning☆215Updated last month
- A simple tutorial of Variational AutoEncoders with Pytorch☆429Updated last year
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆498Updated last year
- torchview: visualize pytorch models☆1,027Updated 8 months ago
- Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)☆1,979Updated last year
- This repository contains a collection of surveys, datasets, papers, and codes, for predictive uncertainty estimation in deep learning mo…☆783Updated last month
- Repository of the journal club "Diffusion Models and Generative Modeling"☆17Updated last year
- Learn fast, scalable, and calibrated measures of uncertainty using neural networks!☆506Updated 4 years ago
- PyTorch implementation of normalizing flow models☆926Updated last year
- VICReg official code base☆553Updated 2 years ago
- Zennit is a high-level framework in Python using PyTorch for explaining/exploring neural networks using attribution methods like LRP.☆239Updated 5 months ago
- 👽 Out-of-Distribution Detection with PyTorch☆332Updated last week
- Quantus is an eXplainable AI toolkit for responsible evaluation of neural network explanations☆636Updated last week
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆721Updated 6 months ago
- Official repository for "Estimating Epistemic and Aleatoric Uncertainty with a Single Model"☆23Updated last year
- Generalized Kraskov Estimator for Normalized Mutual Information☆22Updated last month
- Laplace approximations for Deep Learning.☆532Updated 9 months ago
- [WIP] PyTorch & Hydra Template☆12Updated 3 years ago
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆615Updated last year
- Materials of the Nordic Probabilistic AI School 2022.☆181Updated 3 years ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆434Updated last year
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,552Updated last year
- Simple and clean implementation of Conditional Variational AutoEncoder (cVAE) using PyTorch☆127Updated 2 years ago
- Normalizing flows in PyTorch☆441Updated last month
- Diagrams for visualizing neural network architecture☆1,036Updated 9 months ago
- Lightweight, useful implementation of conformal prediction on real data.☆1,004Updated 2 months ago
- Save matplotlib figures as TikZ/PGFplots for smooth integration into LaTeX.☆275Updated last month
- ☆29Updated 11 months ago
- Example of how to use Weights & Biases on Slurm☆118Updated 3 years ago