lucidrains / En-transformerView external linksLinks
Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network
☆226Jun 2, 2024Updated last year
Alternatives and similar repositories for En-transformer
Users that are interested in En-transformer are comparing it to the libraries listed below
Sorting:
- Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration wit…☆325Aug 28, 2025Updated 5 months ago
- code for the SE3 Transformers paper: https://arxiv.org/abs/2006.10503☆567Jul 10, 2023Updated 2 years ago
- Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…☆171Nov 25, 2022Updated 3 years ago
- Implementation of Denoising Diffusion for protein design, but using the new Equiformer (successor to SE3 Transformers) with some addition…☆57Dec 27, 2022Updated 3 years ago
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of molecules☆58Dec 2, 2020Updated 5 years ago
- A denoising diffusion probabilistic model (DDPM) tailored for conditional generation of protein distograms☆144Apr 20, 2022Updated 3 years ago
- Graph neural network message passing reframed as a Transformer with local attention☆70Dec 24, 2022Updated 3 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆49Jan 27, 2022Updated 4 years ago
- ☆25May 9, 2022Updated 3 years ago
- Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …☆39Aug 3, 2021Updated 4 years ago
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆101Aug 1, 2023Updated 2 years ago
- E(3) Steerable Graph Neural Network☆126Feb 10, 2023Updated 3 years ago
- Implementation of the DDPM + IPA (invariant point attention) for protein generation, as outlined in the paper "Protein Structure and Sequ…☆89Jul 25, 2022Updated 3 years ago
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆140Jan 14, 2022Updated 4 years ago
- An all-atom protein structure dataset for machine learning.☆359Mar 16, 2024Updated last year
- Official implementation of "Generating 3D Molecules for Target Protein Binding" [ICML2022 Long Presentation]☆111Jul 20, 2023Updated 2 years ago
- A library for programmatically generating equivariant layers through constraint solving☆281May 8, 2023Updated 2 years ago
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…☆310Dec 27, 2021Updated 4 years ago
- Implementation for SE(3) diffusion model with application to protein backbone generation☆414Jul 3, 2023Updated 2 years ago
- Implementation of Metaformer, but in an autoregressive manner☆26Jun 21, 2022Updated 3 years ago
- ATOM3D: tasks on molecules in three dimensions☆318Mar 2, 2023Updated 2 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆57Jan 5, 2023Updated 3 years ago
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆36Feb 5, 2021Updated 5 years ago
- ☆524Feb 23, 2022Updated 3 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Apr 6, 2022Updated 3 years ago
- Implementation of Chroma, generative models of protein using DDPM and GNNs, in Pytorch☆160Dec 27, 2022Updated 3 years ago
- A modular framework for neural networks with Euclidean symmetry☆1,218Updated this week
- Fast Discounted Cumulative Sums in PyTorch☆97Aug 28, 2021Updated 4 years ago
- Implementation of Nvidia's NeuralPlexer, for end-to-end differentiable design of functional small-molecules and ligand-binding proteins, …☆52Nov 20, 2023Updated 2 years ago
- [ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs☆269Feb 11, 2025Updated last year
- ChebLieNet, a spectral graph neural network turned equivariant by Riemannian geometry on Lie groups.☆14Aug 20, 2024Updated last year
- Geometric Vector Perceptron --- a rotation-equivariant GNN for learning from biomolecular structure☆162May 1, 2021Updated 4 years ago
- DiffLinker: Equivariant 3D-Conditional Diffusion Model for Molecular Linker Design☆371Apr 17, 2024Updated last year
- Paper list for equivariant neural network☆1,078Aug 19, 2024Updated last year
- ☆34Feb 12, 2025Updated last year
- CLASP - Contrastive Language-Aminoacid Sequence Pretraining☆143Sep 17, 2021Updated 4 years ago
- Implementation of COCO-LM, Correcting and Contrasting Text Sequences for Language Model Pretraining, in Pytorch☆46Mar 3, 2021Updated 4 years ago
- GPT, but made only out of MLPs☆89May 25, 2021Updated 4 years ago
- Implementation of ProteinBERT in Pytorch☆164Aug 10, 2021Updated 4 years ago