Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration with eventual Alphafold2 replication.
☆327Aug 28, 2025Updated 8 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- code for the SE3 Transformers paper: https://arxiv.org/abs/2006.10503☆577Apr 3, 2026Updated last month
- 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 the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…☆289Dec 17, 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 Lie Transformer, Equivariant Self-Attention, in Pytorch☆97Feb 19, 2021Updated 5 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch☆525Dec 6, 2024Updated last year
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pyt…☆77Jun 8, 2021Updated 4 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
- Attempt at reproduction of AlphaFold2☆99Oct 23, 2024Updated last year
- 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
- Demo of diffusion models #DDPM applied to molecular dynamics of small peptide☆14Jan 26, 2024Updated 2 years ago
- E(3) Steerable Graph Neural Network☆128Feb 10, 2023Updated 3 years ago
- Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction☆32Jun 19, 2022Updated 3 years ago
- Implementation of Graph Transformer in Pytorch, for potential use in replicating Alphafold2☆239Jun 22, 2023Updated 2 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Implementation for SE(3) diffusion model with application to protein backbone generation☆419Jul 3, 2023Updated 2 years ago
- To eventually become an unofficial Pytorch implementation / replication of Alphafold2, as details of the architecture get released☆1,632Oct 29, 2022Updated 3 years ago
- Code for Fold2Seq paper from ICML 2021☆50May 25, 2022Updated 3 years ago
- Implementation of Torsional Diffusion for Molecular Conformer Generation (NeurIPS 2022)☆282Feb 10, 2024Updated 2 years ago
- Generative Models for Graph-Based Protein Design☆288Dec 24, 2020Updated 5 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
- A denoising diffusion probabilistic model (DDPM) tailored for conditional generation of protein distograms☆144Apr 20, 2022Updated 4 years ago
- End-To-End Molecular Dynamics (MD) Engine using PyTorch☆700Apr 21, 2026Updated 2 weeks ago
- Official implementation of "Generating 3D Molecules for Target Protein Binding" [ICML2022 Long Presentation]☆111Jul 20, 2023Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Implementation of trRosetta and trDesign for Pytorch, made into a convenient package, for protein structure prediction and design☆83Aug 27, 2021Updated 4 years ago
- An all-atom protein structure dataset for machine learning.☆364Mar 16, 2024Updated 2 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
- A modular framework for neural networks with Euclidean symmetry☆1,250Feb 13, 2026Updated 2 months ago
- ATOM3D: tasks on molecules in three dimensions☆319Mar 2, 2023Updated 3 years ago
- ☆536Feb 23, 2022Updated 4 years ago
- Geometric Vector Perceptrons --- a rotation-equivariant GNN for learning from biomolecular structure☆323Nov 15, 2025Updated 5 months 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 GPT, made only of MLPs, in Jax☆59Jun 23, 2021Updated 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Paper list for equivariant neural network☆1,089Aug 19, 2024Updated last year
- GPT, but made only out of MLPs☆89May 25, 2021Updated 4 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
- Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2☆3,354Dec 16, 2025Updated 4 months ago
- Training neural network potentials☆475Mar 31, 2026Updated last month
- A package to predict protein inter-residue geometries from sequence data☆222Sep 27, 2021Updated 4 years ago
- GearNet and Geometric Pretraining Methods for Protein Structure Representation Learning, ICLR'2023 (https://arxiv.org/abs/2203.06125)☆319Jun 13, 2025Updated 10 months ago