atomicarchitects / equiformer_v2
[ICLR 2024] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations
☆247Updated last week
Alternatives and similar repositories for equiformer_v2:
Users that are interested in equiformer_v2 are comparing it to the libraries listed below
- [ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs☆226Updated last week
- Implementation of Torsional Diffusion for Molecular Conformer Generation (NeurIPS 2022)☆258Updated last year
- GemNet model in PyTorch, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)☆192Updated last year
- Training neural network potentials☆363Updated last week
- Neural Network Force Field based on PyTorch☆262Updated 2 weeks ago
- Code for “From Molecules to Materials Pre-training Large Generalizable Models for Atomic Property Prediction”.☆46Updated 3 months ago
- Code for “FlowMM Generating Materials with Riemannian Flow Matching” and "FlowLLM: Flow Matching for Material Generation with Large Langu…☆108Updated 3 months ago
- GEOM: Energy-annotated molecular conformations☆213Updated 2 years ago
- ☆55Updated 2 months ago
- List of Geometric GNNs for 3D atomic systems☆99Updated 11 months ago
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆57Updated 3 months ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆185Updated this week
- DiffLinker: Equivariant 3D-Conditional Diffusion Model for Molecular Linker Design☆318Updated 10 months ago
- An SE(3)-invariant autoencoder for generating the periodic structure of materials [ICLR 2022]☆268Updated 6 months ago
- Geom3D: Geometric Modeling on 3D Structures, NeurIPS 2023☆117Updated 8 months ago
- Implementation of GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation (ICLR 2022).☆347Updated last year
- jax library for E3 Equivariant Neural Networks☆199Updated 3 weeks ago
- E(3) Steerable Graph Neural Network☆113Updated 2 years ago
- [TMLR 2023] Training and simulating MD with ML force fields☆108Updated 3 months ago
- ☆467Updated 2 years ago
- An object-aware diffusion model for generating chemical reactions☆121Updated 8 months ago
- Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…☆256Updated 2 months ago
- Geometric Latent Diffusion Models for 3D Molecule Generation☆230Updated last year
- G-SchNet - a generative model for 3d molecular structures☆132Updated last year
- DimeNet and DimeNet++ models, as proposed in "Directional Message Passing for Molecular Graphs" (ICLR 2020) and "Fast and Uncertainty-Awa…☆312Updated last year
- MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generation☆97Updated 5 months ago
- Build neural networks for machine learning force fields with JAX☆108Updated last week
- [NeurIPS 2023] The implementation for the paper "Crystal Structure Prediction by Joint Equivariant Diffusion"☆103Updated 7 months ago
- A Euclidean diffusion model for structure-based drug design.☆392Updated 3 weeks ago
- Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pockets☆280Updated last year