lab-cosmo / jax-pmeLinks
Particle-mesh based calculations of long-range interactions in JAX
☆19Updated this week
Alternatives and similar repositories for jax-pme
Users that are interested in jax-pme are comparing it to the libraries listed below
Sorting:
- Training Neural Network potentials through customizable routines in JAX.☆52Updated 2 months ago
- MESS: Modern Electronic Structure Simulations☆39Updated 3 weeks ago
- Particle-mesh based calculations of long-range interactions in PyTorch☆62Updated last week
- Compute neighbor lists for atomistic systems☆60Updated last week
- ☆23Updated last year
- Machine-Learned Interatomic Potential eXploration (mlipx) is designed at BASF for evaluating machine-learned interatomic potentials (MLIP…☆96Updated last week
- Computing representations for atomistic machine learning☆74Updated last week
- Equivariant machine learning interatomic potentials in JAX.☆76Updated this week
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆35Updated last year
- Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics https://arxiv.org/abs/2509.…☆77Updated this week
- Atomistic machine learning models you can use everywhere for everything☆28Updated last week
- ⚛ download and manipulate atomistic datasets☆47Updated last week
- ☆22Updated 5 months ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆14Updated 8 months ago
- ☆12Updated 8 months ago
- [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget.☆44Updated this week
- An implementation of SchNet in JAX and JAX-MD.☆17Updated 3 years ago
- ☆61Updated 2 months ago
- Pretrained model for molecular wavefunctions☆43Updated 3 months ago
- ☆27Updated last month
- MLP training for molecular systems☆54Updated last month
- Collection of tutorials to use the MACE machine learning force field.☆48Updated last year
- Zero Shot Molecular Generation via Similarity Kernels☆28Updated last month
- Machine learning interatomic potential for condensed-phase reactive chemistry☆21Updated last year
- tools for graph-based machine-learning potentials in jax☆25Updated last year
- MACE-OFF23 models☆48Updated 8 months ago
- N-Dimensional MD engine with symmetry group constraints written in C☆50Updated last year
- Force-field-enhanced Neural Networks optimized library☆59Updated 3 weeks ago
- Computing the Committor with the Committor: an Anatomy of the Transition State Ensemble☆12Updated last year
- A flexible and performant framework for training machine learning potentials.☆30Updated this week