The Atomic Architects

All

18 repositories

PriceofFreedom
Public
[GRaM at ICML'24] The Price of Freedom: Exploring Tradeoffs between Expressivity and Computational Efficiency in Equivariant Tensor Products
machine-learning symmetry tensor-product equivariance
Jupyter Notebook
•0•7•0•0•Updated Jan 31, 2025Jan 31, 2025
symphony
Public
[ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation
machine-learning molecular-modeling molecule-generation
Python
•
MIT License
•4•16•0•3•Updated Jan 29, 2025Jan 29, 2025
atomicarchitects.github.io
Public
SCSS
•4•2•0•0•Updated Jan 27, 2025Jan 27, 2025
equiformer
Public
[ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs
machine-learning deep-learning molecular-dynamics pytorch computational-chemistry drug-discovery materials-science force-fields geometric-deep-learning graph-neural-networks
Python
•
MIT License
•43•224•9•1•Updated Jan 22, 2025Jan 22, 2025
datasets
Public
3D datasets useful to the Atomic Architects!
Python
•
MIT License
•0•6•0•0•Updated Jan 12, 2025Jan 12, 2025
DeNS
Public
[TMLR 2024] Generalizing Denoising to Non-Equilibrium Structures Improves Equivariant Force Fields
molecular-dynamics computational-chemistry drug-discovery denoising-autoencoders force-fields geometric-deep-learning graph-neural-networks self-supervised-learning e3nn equivariant-graph-neural-network
Python
•
MIT License
•2•29•0•0•Updated Dec 19, 2024Dec 19, 2024
e3nn_tetris
Public
Simple Tetris Network in Torch 2.0 + C++ inference
Python
•1•4•0•1•Updated Nov 25, 2024Nov 25, 2024
pyspectra
Public
Library for computing, inverting, and visualizing spectra of local atomic environments
Jupyter Notebook
•1•5•0•0•Updated Nov 22, 2024Nov 22, 2024
RelaxedE3NN
Public
[GRaM at ICML'24] Relaxed Equivariant Graph Neural Networks
Jupyter Notebook
•
MIT License
•1•2•0•0•Updated Nov 8, 2024Nov 8, 2024
equivariant-SBS
Public
Jupyter Notebook
•
MIT License
•0•0•0•0•Updated Oct 25, 2024Oct 25, 2024
molmetrics
Public
Metrics for assessing 3D molecular structures.
Python
•
MIT License
•0•6•1•0•Updated Sep 20, 2024Sep 20, 2024
equiformer_v2
Public
[ICLR 2024] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations
machine-learning deep-learning molecular-dynamics pytorch computational-chemistry computational-physics drug-discovery materials-science force-fields geometric-deep-learning
Python
•
MIT License
•32•237•14•1•Updated Jul 31, 2024Jul 31, 2024
BasisLib
Public
Your favorite basis functions for equivariant models in one place. Code borrowed from e3x
Python
•
MIT License
•2•4•2•0•Updated Jun 20, 2024Jun 20, 2024
Symmetry-Breaking-Discovery
Public
Jupyter Notebook
•2•9•0•0•Updated Jun 12, 2024Jun 12, 2024
Equivariant-CNNs-Tutorial
Public
Jupyter Notebook
•1•1•0•0•Updated Apr 20, 2024Apr 20, 2024
FusionFail
Public
Profile showing 3 layers of NequIP using e3nn-jax
Python
•0•0•0•0•Updated Apr 3, 2024Apr 3, 2024
phonax
Public
Jupyter Notebook
•
MIT License
•2•19•1•0•Updated Mar 18, 2024Mar 18, 2024
e3nn
Public
A modular framework for neural networks with Euclidean symmetry
Python
•
Other
•150•3•0•0•Updated Feb 12, 2024Feb 12, 2024