Equivariant machine learning interatomic potentials in JAX.
☆87Feb 10, 2026Updated 3 weeks ago
Alternatives and similar repositories for mace-jax
Users that are interested in mace-jax are comparing it to the libraries listed below
Sorting:
- jax library for E3 Equivariant Neural Networks☆224Aug 25, 2025Updated 6 months ago
- ☆22May 7, 2025Updated 9 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆113Apr 7, 2025Updated 10 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆46Jun 27, 2023Updated 2 years ago
- MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.☆1,082Updated this week
- train and use graph-based ML models of potential energy surfaces☆122Feb 20, 2026Updated 2 weeks ago
- ☆23Nov 1, 2023Updated 2 years ago
- Machine-Learned Interatomic Potential eXploration (mlipx) is designed at BASF for evaluating machine-learned interatomic potentials (MLIP…☆96Jan 28, 2026Updated last month
- Self-describing sparse tensor data format for atomistic machine learning and beyond☆95Updated this week
- Zero Shot Molecular Generation via Similarity Kernels☆28Aug 27, 2025Updated 6 months ago
- GRACE models and gracemaker (as implemented in TensorPotential package)☆86Updated this week
- Particle-mesh based calculations of long-range interactions in JAX☆23Feb 25, 2026Updated last week
- Build neural networks for machine learning force fields with JAX☆133Jun 2, 2025Updated 9 months ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆368Feb 17, 2026Updated 2 weeks ago
- Allegro is an open-source code for building highly scalable and accurate equivariant deep learning interatomic potentials☆463Feb 23, 2026Updated last week
- Training neural network potentials☆468Feb 25, 2026Updated last week
- Compute neighbor lists for atomistic systems☆74Feb 27, 2026Updated last week
- Alchemical machine learning interatomic potentials☆34Nov 8, 2024Updated last year
- Code for automated fitting of machine learned interatomic potentials.☆138Updated this week
- MACE foundation models (MP, OMAT, mh-1)☆204Feb 23, 2026Updated last week
- ☆11Sep 16, 2024Updated last year
- Library for Crystal Symmetry in Rust☆70Updated this week
- Pretrained model for molecular wavefunctions☆55Jan 16, 2026Updated last month
- pytorch implementation of dftd2 & dftd3 (not actively maintained)☆92Jan 13, 2026Updated last month
- Quick Uncertainty and Entropy via STructural Similarity☆56Feb 23, 2026Updated last week
- A modular framework for neural networks with Euclidean symmetry☆1,223Feb 13, 2026Updated 3 weeks ago
- MESS: Modern Electronic Structure Simulations☆43Sep 26, 2025Updated 5 months ago
- ☆102Feb 23, 2026Updated last week
- CUDA implementations of MACE models☆23Aug 19, 2025Updated 6 months ago
- GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine l…☆112Feb 13, 2024Updated 2 years ago
- scalable molecular simulation☆140Oct 13, 2025Updated 4 months ago
- ☆43Updated this week
- ☆120Feb 10, 2026Updated 3 weeks ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆61Jan 7, 2026Updated last month
- An interactive structure/property explorer for materials and molecules☆173Feb 23, 2026Updated last week
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆130Feb 26, 2026Updated last week
- A Python software package for saddle point optimization and minimization of atomic systems.☆136Jan 19, 2026Updated last month
- DMFF (Differentiable Molecular Force Field) is a Jax-based python package that provides a full differentiable implementation of molecular…☆187Jan 15, 2026Updated last month
- NequIP is a code for building E(3)-equivariant interatomic potentials☆872Updated this week