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.☆59Updated 4 months ago
- MESS: Modern Electronic Structure Simulations☆43Updated 2 months ago
- Equivariant machine learning interatomic potentials in JAX.☆81Updated this week
- ☆23Updated 2 years ago
- Compute neighbor lists for atomistic systems☆68Updated 2 weeks ago
- Particle-mesh based calculations of long-range interactions in PyTorch☆65Updated last week
- Computing representations for atomistic machine learning☆74Updated 3 weeks ago
- Machine-Learned Interatomic Potential eXploration (mlipx) is designed at BASF for evaluating machine-learned interatomic potentials (MLIP…☆96Updated 2 weeks ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆36Updated last year
- ☆12Updated 10 months ago
- ⚛ download and manipulate atomistic datasets☆48Updated last month
- Atomistic machine learning models you can use everywhere for everything☆32Updated this week
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆14Updated last month
- ☆22Updated 7 months ago
- tools for graph-based machine-learning potentials in jax☆26Updated last year
- 🌟 [NeurIPS '25 Spotlight] Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics …☆84Updated this week
- ☆33Updated last week
- An implementation of SchNet in JAX and JAX-MD.☆17Updated 3 years ago
- ☆32Updated 2 months ago
- ☆33Updated last week
- ☆65Updated 2 weeks ago
- an interface to semi-empirical quantum chemistry methods implemented with pytorch☆71Updated this week
- Train, fine-tune, and manipulate machine learning models for atomistic systems☆50Updated this week
- [ICLR 2025] Official Implementation of "Towards Fast, Specialized Machine Learning Force Fields: Distilling Foundation Models via Energy …☆21Updated 8 months ago
- train and use graph-based ML models of potential energy surfaces☆117Updated last week
- Zero Shot Molecular Generation via Similarity Kernels☆28Updated 3 months ago
- N-Dimensional MD engine with symmetry group constraints written in C☆50Updated last year
- MACE-OFF23 models☆56Updated 10 months ago
- MLP training for molecular systems☆55Updated 2 weeks ago
- Efficient And Fully Differentiable Extended Tight-Binding☆111Updated this week