ACEsuit / mace-jaxLinks
Equivariant machine learning interatomic potentials in JAX.
☆83Updated this week
Alternatives and similar repositories for mace-jax
Users that are interested in mace-jax are comparing it to the libraries listed below
Sorting:
- ☆23Updated 2 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆46Updated 2 years ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆36Updated last year
- MESS: Modern Electronic Structure Simulations☆43Updated 4 months ago
- ☆22Updated 8 months ago
- [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget and [arXiv'26] Phonon fine-tuning (PFT)☆64Updated this week
- Particle-mesh based calculations of long-range interactions in JAX☆23Updated this week
- Compute neighbor lists for atomistic systems☆73Updated this week
- Computing representations for atomistic machine learning☆75Updated last week
- Training Neural Network potentials through customizable routines in JAX.☆62Updated 6 months ago
- CUDA implementations of MACE models☆23Updated 5 months ago
- MESS: Modern Electronic Structure Simulations☆20Updated last year
- Build neural networks for machine learning force fields with JAX☆132Updated 8 months ago
- Official repository for the paper "Uncertainty-biased molecular dynamics for learning uniformly accurate interatomic potentials".☆21Updated last year
- Reference implementation of "SpookyNet: Learning force fields with electronic degrees of freedom and nonlocal effects"☆85Updated 3 years ago
- ☆32Updated 4 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆28Updated 11 months ago
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆39Updated last year
- Atomistic machine learning models you can use everywhere for everything☆33Updated last week
- Generate and predict molecular electron densities with Euclidean Neural Networks☆49Updated 2 years ago
- A library for building equivariant neural networks and a zoo of implementations & examples.☆31Updated 3 years ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆15Updated 2 months ago
- Deprecated - see `pair_nequip_allegro`☆44Updated 9 months ago
- N-Dimensional MD engine with symmetry group constraints written in C☆49Updated last year
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆57Updated 4 months ago
- Introductory lectures in atomistic machine learning☆23Updated 6 months ago
- [npj Comp. Mat.] Higher-order equivariant neural networks for charge density prediction in materials☆71Updated 11 months ago
- ☆17Updated last year
- [ICLR 2025] Official Implementation of "Towards Fast, Specialized Machine Learning Force Fields: Distilling Foundation Models via Energy …☆21Updated 9 months ago
- an interface to semi-empirical quantum chemistry methods implemented with pytorch☆72Updated last week