lucidrains / se3-transformer-pytorchView external linksLinks
Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration with eventual Alphafold2 replication.
☆325Aug 28, 2025Updated 5 months ago
Alternatives and similar repositories for se3-transformer-pytorch
Users that are interested in se3-transformer-pytorch are comparing it to the libraries listed below
Sorting:
- code for the SE3 Transformers paper: https://arxiv.org/abs/2006.10503☆567Jul 10, 2023Updated 2 years ago
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆226Jun 2, 2024Updated last year
- 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 the DDPM + IPA (invariant point attention) for protein generation, as outlined in the paper "Protein Structure and Sequ…☆89Jul 25, 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
- E(3) Steerable Graph Neural Network☆126Feb 10, 2023Updated 3 years ago
- Attempt at reproduction of AlphaFold2☆98Oct 23, 2024Updated last year
- A denoising diffusion probabilistic model (DDPM) tailored for conditional generation of protein distograms☆144Apr 20, 2022Updated 3 years ago
- Demo of diffusion models #DDPM applied to molecular dynamics of small peptide☆14Jan 26, 2024Updated 2 years ago
- Implementation for SE(3) diffusion model with application to protein backbone generation☆414Jul 3, 2023Updated 2 years ago
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆16Aug 3, 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
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of molecules☆58Dec 2, 2020Updated 5 years ago
- End-To-End Molecular Dynamics (MD) Engine using PyTorch☆680Jan 7, 2026Updated last month
- Geometric Vector Perceptrons --- a rotation-equivariant GNN for learning from biomolecular structure☆317Nov 15, 2025Updated 3 months 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
- Paper list for equivariant neural network☆1,078Aug 19, 2024Updated last year
- Generative Models for Graph-Based Protein Design☆284Dec 24, 2020Updated 5 years ago
- An all-atom protein structure dataset for machine learning.☆359Mar 16, 2024Updated last year
- ATOM3D: tasks on molecules in three dimensions☆318Mar 2, 2023Updated 2 years ago
- Official implementation of "Generating 3D Molecules for Target Protein Binding" [ICML2022 Long Presentation]☆111Jul 20, 2023Updated 2 years ago
- A modular framework for neural networks with Euclidean symmetry☆1,218Updated this week
- Implementation of Torsional Diffusion for Molecular Conformer Generation (NeurIPS 2022)☆280Feb 10, 2024Updated 2 years ago
- Implementation and replication of ProGen, Language Modeling for Protein Generation, in Jax☆113Sep 8, 2021Updated 4 years ago
- ☆29Jul 10, 2020Updated 5 years ago
- Code for Fold2Seq paper from ICML 2021☆50May 25, 2022Updated 3 years ago
- Research repository for the proposed equivariant graph attention network that operates on large biomolecules proposed by Le et al. (2022)☆20Dec 1, 2022Updated 3 years ago
- GPT, but made only out of MLPs☆89May 25, 2021Updated 4 years ago
- Official code repository for EquiFold: Protein Structure Prediction with a Novel Coarse-Grained Structure Representation☆127Jan 8, 2023Updated 3 years ago
- Universal Structure Alignment of Monomeric and Complex Structure of Nucleic Acids and Proteins☆174Feb 4, 2026Updated last week
- pytorch implementation of trDesign☆44Mar 19, 2021Updated 4 years ago
- Local Attention - Flax module for Jax☆22May 26, 2021Updated 4 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
- ☆524Feb 23, 2022Updated 3 years ago
- Implementation of ProteinBERT in Pytorch☆164Aug 10, 2021Updated 4 years ago
- ☆116Dec 7, 2022Updated 3 years ago
- Training neural network potentials☆466Jan 26, 2026Updated 3 weeks ago
- GearNet and Geometric Pretraining Methods for Protein Structure Representation Learning, ICLR'2023 (https://arxiv.org/abs/2203.06125)☆315Jun 13, 2025Updated 8 months ago
- Geometric Vector Perceptron --- a rotation-equivariant GNN for learning from biomolecular structure☆162May 1, 2021Updated 4 years ago