e3nn

There are 1 repository under e3nn topic.

chaitjo / geometric-gnn-dojo
Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks.
geometric-deep-learning graph-neural-networks equivariance equivariant-networks geometric-graphs graph-isomorphism pytorch pytorch-geometric e3nn
Language:Jupyter Notebook 433
atomicarchitects / equiformer
[ICLR'23 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs
catalyst-design computational-chemistry deep-learning drug-discovery equivariant-graph-neural-network force-fields interatomic-potentials machine-learning molecular-dynamics pytorch ai-for-science equivariant-neural-networks graph-neural-networks e3nn geometric-deep-learning materials-science
Language:Python 177
atomicarchitects / equiformer_v2
[ICLR'24] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations
catalyst-design computational-chemistry computational-physics deep-learning drug-discovery equivariant-graph-neural-network equivariant-neural-networks force-fields interatomic-potentials machine-learning molecular-dynamics pytorch ai-for-science graph-neural-networks e3nn geometric-deep-learning materials-science
Language:Python 151