Equiformer

Machine Learning for Physics
Maintenance light282updated 1 year ago
Python
MIT

Equivariant graph attention Transformer (ICLR2023)

README

Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs Paper | OpenReview | Poster | Slides This repository contains the official PyTorch implementation of the work "Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs" (ICLR 2023 Spotlight). Additionally, in our subsequent work, we find that we can generalize self-supervised learning similar to BERT, which we call DeNS (Denoising Non-Equilibrium Structures), to 3D atomistic systems to improve the…

Source attribution

  • GitHubgithub.com/atomicarchitects/equiformer
  • Awesome AI for Sciencegithub.com/atomicarchitects/equiformer

Related resources

E(3)-equivariant neural network interatomic potentials achieving DFT accuracy with up to 1000× less training data than invariant models, foundational architecture behind MACE and Allegro (Harvard, MIT, Nature Communications 2022)

9144 days ago
Python
MIT

OpenChem is a deep learning toolkit for Computational Chemistry with PyTorch backend.

7452 years ago
Python
MIT

Deep learning library for Chemistry based on Tensorflow

6.8K2 months ago
Python
MIT

Provides functionality for producing geometric representations of protein and RNA structures, and biological interaction networks.

1.2K2 days ago
Jupyter Notebook
MIT

Cross-platform library for differentiable programming of quantum computers with automatic differentiation, enabling hybrid quantum-classical machine learning for quantum chemistry, quantum physics, and NISQ algorithm research (Xanadu, 3k+ stars)

3.2K2 days ago
Python
Apache-2.0

Open-source framework for building physics-ML models at scale (renamed from Modulus, 2025)

2.8K2 days ago
Python
Apache-2.0