tbepler / spatial-VAE
Source code for "Explicitly disentangling image content from translation and rotation with spatial-VAE" - NeurIPS 2019
☆65Updated 5 years ago
Alternatives and similar repositories for spatial-VAE:
Users that are interested in spatial-VAE are comparing it to the libraries listed below
- This code accompanies "Differentiable probabilistic models of scientific imaging with the Fourier slice theorem", UAI 2019☆37Updated 5 years ago
- VAEs with Lie Group latent space☆97Updated 3 years ago
- Energy-based models for atomic-resolution protein conformations☆96Updated 2 years ago
- ☆122Updated 2 years ago
- ☆42Updated last year
- ☆27Updated 4 years ago
- Euclidean Neural Networks☆190Updated 4 years ago
- Code for the arXiv preprint "Disentangling images with Lie group transformations and sparse coding" (2020).☆13Updated 3 years ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆88Updated 4 years ago
- Gaussian Process Prior Variational Autoencoder☆82Updated 6 years ago
- Learning to recover orientations from projections in single-particle cryo-EM☆15Updated 3 years ago
- ICML 2020 Paper: Latent Variable Modelling with Hyperbolic Normalizing Flows☆53Updated 2 years ago
- Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…☆158Updated 2 years ago
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pyt…☆72Updated 3 years ago
- Lie Algebra Convolutional Network implementation☆42Updated 3 years ago
- ☆38Updated 4 years ago
- GraphNVP: An Invertible Flow Model for Generating Molecular Graphs☆92Updated 2 years ago
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆115Updated last year
- CPAB Transformations: finite-dimensional spaces of simple, fast, and highly-expressive diffeomorphisms derived from parametric, continuou…☆48Updated 3 years ago
- Equivariant Transformer (ET) layers are image-to-image mappings that incorporate prior knowledge on invariances with respect to continuo…☆89Updated 5 years ago
- Learning protein structure with a differentiable simulator☆27Updated 5 years ago
- Pytorch implementations of generative models: VQVAE2, AIR, DRAW, InfoGAN, DCGAN, SSVAE☆91Updated 4 years ago
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆129Updated 3 years ago
- ☆15Updated 3 years ago
- Pytorch implementation of the Power Spherical distribution☆74Updated 7 months ago
- Code for "Variational Autoencoder with Learned Latent Structure"☆33Updated 3 years ago
- ☆157Updated 5 years ago
- ☆69Updated last year
- Convolutional variational autoencoder in PyTorch☆46Updated 6 years ago
- ☆89Updated 5 years ago