Particle-mesh based calculations of long-range interactions in JAX
☆28Apr 18, 2026Updated 3 weeks ago
Alternatives and similar repositories for jax-pme
Users that are interested in jax-pme are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- tools for graph-based machine-learning potentials in jax☆27Apr 9, 2024Updated 2 years ago
- Compute neighbor lists for atomistic systems☆79Updated this week
- Computing representations for atomistic machine learning☆80Apr 2, 2026Updated last month
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆120Apr 22, 2026Updated 2 weeks ago
- Particle-mesh based calculations of long-range interactions in PyTorch☆79May 3, 2026Updated last week
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Self-describing sparse tensor data format for atomistic machine learning and beyond☆98Updated this week
- ☆50Mar 27, 2026Updated last month
- ☆22May 7, 2025Updated last year
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated last year
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆53May 4, 2026Updated last week
- Implementation of the Euclidean fast attention (EFA) algorithm☆95Jan 7, 2026Updated 4 months ago
- An implementation of SchNet in JAX and JAX-MD.☆17Apr 5, 2022Updated 4 years ago
- Point Edge Transformer☆34Sep 28, 2025Updated 7 months ago
- ☆23Mar 18, 2024Updated 2 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Equivariant machine learning interatomic potentials in JAX.☆99Apr 25, 2026Updated 2 weeks ago
- ☆128May 3, 2026Updated last week
- Calculation of Wigner symbols and related constants☆13May 13, 2025Updated 11 months ago
- An engine for electrostatic ML embedding for multiscale modelling.☆27Apr 28, 2026Updated last week
- Pretrained model for molecular wavefunctions☆58Apr 21, 2026Updated 2 weeks ago
- DMFF (Differentiable Molecular Force Field) is a Jax-based python package that provides a full differentiable implementation of molecular…☆195Apr 7, 2026Updated last month
- Machine Learned Interatomic Potential Tools☆25Apr 27, 2026Updated 2 weeks ago
- Machine-Learned Interatomic Potential eXploration (mlipx) is designed at BASF for evaluating machine-learned interatomic potentials (MLIP…☆103Jan 28, 2026Updated 3 months ago
- SO3krates and Universal Pairwise Force Field for Molecular Simulation☆211Apr 17, 2026Updated 3 weeks ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Transferable Double Exponential non-bonded potential for condensed phase simulations of small molecules☆25Feb 4, 2026Updated 3 months ago
- Zero Shot Molecular Generation via Similarity Kernels☆29Aug 27, 2025Updated 8 months ago
- Graph neural network prediction of electronic Hamiltonians in atomic orbital representation with many body messages☆32Apr 7, 2026Updated last month
- MESS: Modern Electronic Structure Simulations☆43Sep 26, 2025Updated 7 months ago
- A collection of simulation recipes for the atomic-scale modeling of materials and molecules☆49May 4, 2026Updated last week
- JAX-ReaxFF: A Gradient Based Framework for Extremely Fast Optimization of Reactive Force Fields☆79Sep 23, 2024Updated last year
- ☆14Feb 11, 2025Updated last year
- SOAP (Smooth Overlap of Atomic Positions) implementation with numpy☆11May 11, 2018Updated 8 years ago
- The integrated tempering sampling as a bias in PLUMED2☆11Jan 15, 2021Updated 5 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Train, fine-tune, and manipulate machine learning models for atomistic systems☆68Updated this week
- Build neural networks for machine learning force fields with JAX☆136Jun 2, 2025Updated 11 months ago
- MACE-OFF23 models☆65Jan 29, 2025Updated last year
- Training Neural Network potentials through customizable routines in JAX.☆68Feb 10, 2026Updated 3 months ago
- Python implementation of the umbrella integration method for potential of mean force (PMF) calculations☆17Feb 4, 2019Updated 7 years ago
- Multi-language library for the calculation of spherical harmonics in Cartesian coordinates☆96May 4, 2026Updated last week
- Differentiable Markov Chain Monte Carlo☆15Mar 23, 2024Updated 2 years ago