pytorch-vae

A Collection of Variational Autoencoders (VAE) in PyTorch.

KAE : KAN-based AutoEncoder (AE, VAE, VQ-VAE, RVQ, etc.)

Official PyTorch implementation of A Quaternion-Valued Variational Autoencoder (QVAE).

Unofficial implementation of the paper Semantic Deep Face Models by Prashanth Chandran et al. (2020)

Example of replacing MLP with KAN in autoencoder(AE) and variational autoencoder(VAE)