Training Neural Network potentials through customizable routines in JAX.
☆63Feb 10, 2026Updated 2 weeks ago
Alternatives and similar repositories for chemtrain
Users that are interested in chemtrain are comparing it to the libraries listed below
Sorting:
- ☆16Feb 17, 2025Updated last year
- Implementation of Differentiable Molecular Simulations with torchMD.☆16Oct 9, 2023Updated 2 years ago
- An implementation of SchNet in JAX and JAX-MD.☆17Apr 5, 2022Updated 3 years ago
- Zero Shot Molecular Generation via Similarity Kernels☆28Aug 27, 2025Updated 6 months ago
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated 11 months ago
- ☆22May 7, 2025Updated 9 months ago
- An overview of literature that discusses the use of machine learning for atomistic simulations☆44Apr 11, 2023Updated 2 years ago
- Particle-mesh based calculations of long-range interactions in JAX☆23Updated this week
- ☆12Feb 15, 2026Updated 2 weeks ago
- PyTorch/PyG implementation of M3GNet☆11Sep 11, 2023Updated 2 years ago
- Force-field-enhanced Neural Networks optimized library☆81Feb 9, 2026Updated 3 weeks ago
- Cross-platform Optimizer for ML Interatomic Potentials☆23Aug 31, 2025Updated 6 months ago
- Transferable Double Exponential non-bonded potential for condensed phase simulations of small molecules☆25Feb 4, 2026Updated 3 weeks ago
- ☆31Jul 8, 2025Updated 7 months ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆37Feb 29, 2024Updated 2 years ago
- MESS: Modern Electronic Structure Simulations☆43Sep 26, 2025Updated 5 months ago
- jax library for E3 Equivariant Neural Networks☆224Aug 25, 2025Updated 6 months ago
- Stochastic Gradient MCMC for Jax☆18May 19, 2025Updated 9 months ago
- Julia tools for reading Crystallographic Information Framework (CIF) files and dictionaries☆15Feb 2, 2026Updated last month
- scalable molecular simulation☆140Oct 13, 2025Updated 4 months ago
- A collection of simulation recipes for the atomic-scale modeling of materials and molecules☆35Feb 23, 2026Updated last week
- This repository contains the source code for Bayesian Learned Interatomic Potentials (BLIP)☆31Aug 20, 2025Updated 6 months ago
- A Python library for constructing polymer topologies and coordinates☆18Sep 23, 2025Updated 5 months ago
- Standalone charge assignment from Espaloma framework.☆46Oct 10, 2025Updated 4 months ago
- ☆42Feb 13, 2026Updated 2 weeks ago
- Compute neighbor lists for atomistic systems☆74Updated this week
- A unified framework for machine learning collective variables for enhanced sampling simulations☆135Feb 19, 2026Updated last week
- [ICLR 2025] Official Implementation of "Towards Fast, Specialized Machine Learning Force Fields: Distilling Foundation Models via Energy …☆22Apr 17, 2025Updated 10 months ago
- Jax / Haiku implementation of DimeNet++.☆18Mar 31, 2022Updated 3 years ago
- [ICLR 2024] Official Implementation of Long-Short-Range Message-Passing☆19Sep 7, 2025Updated 5 months ago
- Pytorch differentiable molecular dynamics☆184Sep 5, 2022Updated 3 years ago
- ☆34Oct 18, 2024Updated last year
- MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.☆1,064Feb 23, 2026Updated last week
- Example to fit parameters and run CG simulations using TorchMD and Schnet☆47Feb 4, 2022Updated 4 years ago
- TrajCast: Force-Free MD Through Autoregressive Equivariant Networks☆60Sep 23, 2025Updated 5 months ago
- OpenMM plugin that implements (an)isotropic polarizable point dipoles and multipoles up to octopoles.☆11Feb 7, 2025Updated last year
- working with crystal structures☆21Jun 28, 2024Updated last year
- MaterialsCoord: infrastructure to benchmark crystal structure coordination algorithms☆20Mar 13, 2023Updated 2 years ago
- Physical validation of molecular simulations☆58Jan 5, 2026Updated last month