lucidrains / logavgexp-torch
Implementation of LogAvgExp for Pytorch
☆32Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for logavgexp-torch
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆35Updated 3 years ago
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 2 years ago
- Another attempt at a long-context / efficient transformer by me☆37Updated 2 years ago
- A convolution-free, transformer-only version of the CycleGAN framework☆32Updated 2 years ago
- ☆21Updated last year
- PyTorch implementation of Contrastive Feature Loss for Image Prediction (AIM Workshop at ICCV 2021)☆53Updated 3 years ago
- JAX implementation ViT-VQGAN☆77Updated 2 years ago
- ☆28Updated 2 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆24Updated last year
- ☆33Updated last year
- A simple Transformer where the softmax has been replaced with normalization☆18Updated 4 years ago
- ☆40Updated last year
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆52Updated last year
- ☆47Updated 2 years ago
- FID computation in Jax/Flax.☆24Updated 4 months ago
- Repository for the PopulAtion Parameter Averaging (PAPA) paper☆26Updated 7 months ago
- Official repository for MaGNET, ICLR 2022☆26Updated 2 years ago
- A simple implementation of a deep linear Pytorch module☆18Updated 4 years ago
- Official PyTorch implementation of Spatial Dependency Networks.☆35Updated 3 years ago
- ImageNet-12k subset of ImageNet-21k (fall11)☆20Updated last year
- ICCV2021 (poster)☆74Updated 3 years ago
- A JAX nn library☆21Updated 8 months ago