zrqiao / torch-gaugeLinks
A light-weight PyTorch extension for equivariant deep learning
☆17Updated 4 months ago
Alternatives and similar repositories for torch-gauge
Users that are interested in torch-gauge are comparing it to the libraries listed below
Sorting:
- GemNet model in TensorFlow, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)☆26Updated 2 years ago
- Contrastive pretraining to learn chemical reaction representations (RxnRep) for downstream tasks.☆33Updated 2 years ago
- Geometric super-resolution for molecular geometries☆41Updated 3 years ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆50Updated 2 years ago
- Data and scripts for comprehensive benchmark of conformer relative energies☆25Updated 5 years ago
- Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.☆30Updated 3 years ago
- Example to fit parameters and run CG simulations using TorchMD and Schnet☆46Updated 3 years ago
- Deep learning for molecules quantum chemistry properties prediction☆41Updated 4 years ago
- A graph neural network for the prediction of bond dissociation energies for molecules of any charge.☆63Updated 2 years ago
- A library for building equivariant neural networks and a zoo of implementations & examples.☆32Updated 2 years ago
- Repository for Chemical Perception Sampling Tools☆21Updated 11 months ago
- Force-field-enhanced Neural Networks optimized library☆57Updated last week
- ☆19Updated 2 years ago
- Robust NN MD simulator☆20Updated last year
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆23Updated 4 months ago
- A Newtonian message passing network for deep learning of interatomic potentials and forces☆44Updated last month
- [ICML'24] Adsorbate Placement via Conditional Denoising Diffusion☆18Updated last year
- Implementing PaiNN in Pytorch Geometric☆14Updated 3 years ago
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆35Updated 6 months ago
- Supplementary Data for "A graph representation of molecular ensembles for polymer property prediction"☆20Updated 2 years ago
- ☆14Updated 9 months ago
- [TMLR 2023] Training and simulating MD with ML force fields☆111Updated 8 months ago
- Repository of Quantum Datasets Publicly Available☆49Updated 3 weeks ago
- AIMNet2: Fast, accurate and transferable neural network interatomic potential☆18Updated 8 months ago
- Diffusion model for transition state prediction☆35Updated last year
- Mindless molecule generator in a Python package.☆37Updated 2 weeks ago
- Reinforced dynamics☆49Updated 3 months ago
- Automated reaction discovery and dataset generation with the growing string method☆20Updated 5 years ago
- Computational chemistry benchmark data sets for non-covalent interactions☆29Updated 6 months ago
- Conformer-Rotamer Ensemble Sampling Tool based on the xtb Semiempirical Extended Tight-Binding Program Package☆36Updated 2 weeks ago