zrqiao / torch-gauge
A light-weight PyTorch extension for equivariant deep learning
☆15Updated this week
Alternatives and similar repositories for torch-gauge:
Users that are interested in torch-gauge are comparing it to the libraries listed below
- Contrastive pretraining to learn chemical reaction representations (RxnRep) for downstream tasks.☆32Updated 2 years ago
- Geometric super-resolution for molecular geometries☆39Updated 2 years ago
- GemNet model in TensorFlow, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)☆25Updated last year
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆45Updated last year
- Molecular mechanics systems and simulation data☆15Updated 8 months ago
- Example to fit parameters and run CG simulations using TorchMD and Schnet☆43Updated 3 years ago
- https://arxiv.org/abs/2102.11439☆20Updated 3 years ago
- Synthetic coordinates for GNNs, as proposed in "Directional Message Passing on Molecular Graphs via Synthetic Coordinates" (NeurIPS 2021)☆29Updated last year
- The official implementation of Energy-Inspired Molecular Conformation Optimization (ICLR 2022)☆36Updated 2 years ago
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆17Updated 2 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆16Updated this week
- Data and scripts for comprehensive benchmark of conformer relative energies☆23Updated 4 years ago
- Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.☆29Updated 2 years ago
- This repo contains the datasets used for the paper The Design Space of E(3)-Equivariant Atom-Centered Interatomic Potentials.☆18Updated 2 years ago
- Deep Supervised Graph Partitioning Model☆14Updated 3 years ago
- Implementing PaiNN in Pytorch Geometric☆13Updated 2 years ago
- AIMNet2: Fast, accurate and transferable neural network interatomic potential☆18Updated 4 months ago
- Molecular Property Prediction using GP with a SOAP kernel☆21Updated 4 years ago
- Robust NN MD simulator☆20Updated last year
- Deep learning for molecules quantum chemistry properties prediction☆38Updated 3 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆35Updated last year
- Implementation of Differentiable Molecular Simulations with torchMD.☆14Updated last year
- Learning Neural Generative Dynamics for Molecular Conformation Generation (ICLR 2021)☆22Updated 3 years ago
- An End-to-End Framework for Molecular Conformation Generation via Bilevel Programming (ICML'21)☆51Updated 3 years ago
- A graph neural network for the prediction of bond dissociation energies for molecules of any charge.☆56Updated last year
- Repository for Chemical Perception Sampling Tools☆19Updated 6 months ago
- Code for performing adversarial attacks on atomistic systems using NN potentials☆36Updated 2 years ago
- Generative model for molecular distance geometry☆38Updated last year
- A library for building equivariant neural networks and a zoo of implementations & examples.☆31Updated 2 years ago
- Jax / Haiku implementation of DimeNet++.☆18Updated 2 years ago