Playground

Implementations of various neural network architectures using PyTorch Lightning.

Simple experiments run on Google Colab notebooks.

Implementations

List of implementations and their status.

CNNs

• AlexNet
• ResNet
• VGGNet
• DenseNet

Autoencoders

• Vanilla autoencoder
• Variational autoencoders (VAE)
• Vector-Quantized VAE (VQ-VAE)
• Nouveau VAEs (NVAE)
• SwapVAE

GANs

• Vanilla GAN

Federated Learning

• Federated Averaging (FedAvg)
• Federated SGD (FedSGD)

Resources

• PyTorch Lightning