Equivariant machine learning interatomic potentials in JAX.
☆107Apr 25, 2026Updated 3 months ago
Alternatives and similar repositories for mace-jax
Users that are interested in mace-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆22May 7, 2025Updated last year
- jax library for E3 Equivariant Neural Networks☆233Apr 1, 2026Updated 3 months ago
- MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.☆1,293Updated this week
- CUDA implementations of MACE models☆23Aug 19, 2025Updated 11 months ago
- Training Neural Network potentials through customizable routines in JAX.☆73Jun 30, 2026Updated 3 weeks ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Higher order equivariant graph neural networks for 3D point clouds☆49Jun 27, 2023Updated 3 years ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆125Jul 2, 2026Updated 3 weeks ago
- Compute neighbor lists for atomistic systems☆83Updated this week
- MACE foundation models (MP, OMAT, mh-1)☆285Jul 12, 2026Updated 2 weeks ago
- ☆11Sep 16, 2024Updated last year
- Zero Shot Molecular Generation via Similarity Kernels☆29Aug 27, 2025Updated 11 months ago
- train and use graph-based ML models of potential energy surfaces☆127Jul 13, 2026Updated 2 weeks ago
- ☆25Nov 1, 2023Updated 2 years ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆416Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆45Updated this week
- ☆57Updated this week
- An ASE-friendly implementation of the amorphous-to-crystalline (a2c) workflow.☆18Oct 19, 2025Updated 9 months ago
- Pretrained model for molecular wavefunctions☆62Jul 3, 2026Updated 3 weeks ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆161Updated this week
- Self-describing sparse tensor data format for atomistic machine learning and beyond☆106Updated this week
- MACE-OFF23 models☆75Jan 29, 2025Updated last year
- Machine-Learned Interatomic Potential eXploration (mlipx) is designed at BASF for evaluating machine-learned interatomic potentials (MLIP…☆106Jan 28, 2026Updated 6 months ago
- SevenNet - a graph neural network interatomic potential package supporting efficient multi-GPU parallel molecular dynamics simulations.☆261Jul 22, 2026Updated last week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Alchemical machine learning interatomic potentials☆36Nov 8, 2024Updated last year
- ☆18Feb 26, 2026Updated 5 months ago
- 🌟 [NeurIPS '25 Spotlight] Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics …☆105Jul 2, 2026Updated 3 weeks ago
- Enhanced sampling in the representation space of atomistic neural networks.☆15Updated this week
- pytorch implementation of dftd2 & dftd3 (not actively maintained)☆107Mar 11, 2026Updated 4 months ago
- Collection of tutorials to use the MACE machine learning force field.☆69Jul 13, 2026Updated 2 weeks ago
- [ICML'26] Phonon fine-tuning (PFT) and [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget☆76Apr 5, 2026Updated 3 months ago
- Particle-mesh based calculations of long-range interactions in JAX☆30Updated this week
- Machine Learning Interatomic Potentials with the Atomic Cluster Expansion☆77Jun 29, 2026Updated 3 weeks ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ML Performance and Extrapolation Guide (reference deployment https://ml-peg.stfc.ac.uk)☆54Updated this week
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆43Mar 1, 2025Updated last year
- MESS: Modern Electronic Structure Simulations☆43May 9, 2026Updated 2 months ago
- SO3krates and Universal Pairwise Force Field for Molecular Simulation☆227Apr 17, 2026Updated 3 months ago
- Code for automated fitting of machine learned interatomic potentials.☆155Updated this week
- Benchmarking foundation Machine Learning Potentials with Lattice Thermal Conductivity from Anharmonic Phonons☆16Oct 30, 2024Updated last year
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)☆124Updated this week