lab-cosmo / jax-pmeView external linksLinks
Particle-mesh based calculations of long-range interactions in JAX
☆23Updated 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:
- tools for graph-based machine-learning potentials in jax☆26Apr 9, 2024Updated last year
- Particle-mesh based calculations of long-range interactions in PyTorch☆75Jan 29, 2026Updated 2 weeks ago
- Compute neighbor lists for atomistic systems☆73Updated this week
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆110Apr 7, 2025Updated 10 months ago
- Computing representations for atomistic machine learning☆77Feb 4, 2026Updated last week
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated 10 months ago
- ☆22May 7, 2025Updated 9 months ago
- Self-describing sparse tensor data format for atomistic machine learning and beyond☆93Feb 6, 2026Updated last week
- ☆49Updated this week
- An engine for electrostatic ML embedding for multiscale modelling.☆26Jan 14, 2026Updated last month
- ☆118Feb 8, 2026Updated last week
- Implementation of the Euclidean fast attention (EFA) algorithm☆60Jan 7, 2026Updated last month
- 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 5 months ago
- Pretrained model for molecular wavefunctions☆55Jan 16, 2026Updated 3 weeks ago
- MESS: Modern Electronic Structure Simulations☆43Sep 26, 2025Updated 4 months ago
- DMFF (Differentiable Molecular Force Field) is a Jax-based python package that provides a full differentiable implementation of molecular…☆188Jan 15, 2026Updated last month
- Machine-Learned Interatomic Potential eXploration (mlipx) is designed at BASF for evaluating machine-learned interatomic potentials (MLIP…☆96Jan 28, 2026Updated 2 weeks ago
- MACE-OFF23 models☆59Jan 29, 2025Updated last year
- Train, fine-tune, and manipulate machine learning models for atomistic systems☆54Updated this week
- Transferable Double Exponential non-bonded potential for condensed phase simulations of small molecules☆25Feb 4, 2026Updated last week
- A flexible and performant framework for training machine learning potentials.☆32Updated this week
- Machine Learned Interatomic Potential Tools☆24Updated this week
- JAX-ReaxFF: A Gradient Based Framework for Extremely Fast Optimization of Reactive Force Fields☆74Sep 23, 2024Updated last year
- ☆40Jan 23, 2026Updated 3 weeks ago
- Training Neural Network potentials through customizable routines in JAX.☆61Jul 30, 2025Updated 6 months ago
- ☆22Mar 18, 2024Updated last year
- ☆12Feb 11, 2025Updated last year
- Constant pH simulation with OpenMM☆18Mar 26, 2025Updated 10 months ago
- SO3krates and Universal Pairwise Force Field for Molecular Simulation☆201Updated this week
- A collection of simulation recipes for the atomic-scale modeling of materials and molecules☆34Feb 8, 2026Updated last week
- A set of tools for analyzing molecular dynamics simulations☆12Feb 5, 2026Updated last week
- The integrated tempering sampling as a bias in PLUMED2☆11Jan 15, 2021Updated 5 years ago
- Calculation of Wigner symbols and related constants☆11May 13, 2025Updated 9 months ago
- Equivariant machine learning interatomic potentials in JAX.☆83Feb 2, 2026Updated 2 weeks ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆51Jun 13, 2023Updated 2 years ago
- DeePMD-kit plugin for various graph neural network models☆52Updated this week
- MACE foundation models (MP, OMAT, mh-1)☆200Nov 19, 2025Updated 2 months ago
- general purpose coarse-grained molecular dynamics simulation package☆35Jan 27, 2023Updated 3 years ago