Variational Autoencoder (VAE) with Discrete Distribution using Gumbel Softmax

Theory and PyTorch Implementation

Alexey Kravets
Towards Data Science
17 min readAug 9, 2023

--

https://unsplash.com/photos/sbVu5zitZt0

Since this article is going to be extensive, I will provide the reader with an index for better navigation:

  1. Introduction
  2. Brief Introduction to Variational Autoencoders (VAEs)
  3. Kullback–Leibler (KL)…

--

--