ACEsuit / mace-jax
Equivariant machine learning interatomic potentials in JAX.
☆72Updated last year
Alternatives and similar repositories for mace-jax:
Users that are interested in mace-jax are comparing it to the libraries listed below
- Higher order equivariant graph neural networks for 3D point clouds☆36Updated last year
- MESS: Modern Electronic Structure Simulations☆27Updated 5 months ago
- ☆23Updated last year
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- Computing representations for atomistic machine learning☆67Updated last month
- Reference implementation of "SpookyNet: Learning force fields with electronic degrees of freedom and nonlocal effects"☆71Updated 2 years ago
- ☆21Updated 11 months ago
- ☆17Updated 2 years ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆44Updated 2 weeks ago
- Official implementation of DeepDFT model☆73Updated 2 years ago
- Build neural networks for machine learning force fields with JAX☆111Updated last month
- This repo contains the datasets used for the paper The Design Space of E(3)-Equivariant Atom-Centered Interatomic Potentials.☆18Updated 2 years ago
- Generate and predict molecular electron densities with Euclidean Neural Networks☆46Updated last year
- MESS: Modern Electronic Structure Simulations☆19Updated 6 months ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆46Updated last year
- Jax / Haiku implementation of DimeNet++.☆18Updated 2 years ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆45Updated this week
- python workflow toolkit☆37Updated last month
- Compute neighbor lists for atomistic systems☆49Updated 3 weeks ago
- JAX-ReaxFF: A Gradient Based Framework for Extremely Fast Optimization of Reactive Force Fields☆62Updated 6 months ago
- CUDA implementations of MACE models☆14Updated 2 months ago
- ☆21Updated last year
- train and use graph-based ML models of potential energy surfaces☆78Updated this week
- LAMMPS pair style for Allegro deep learning interatomic potentials with parallelization support☆41Updated 5 months ago
- ☆24Updated 11 months ago
- ☆45Updated 9 months ago
- A repository to hold installation recipes and data files for the PET-MAD universal machine-learning interatomic potential☆36Updated this week
- Official repository for the paper "Uncertainty-biased molecular dynamics for learning uniformly accurate interatomic potentials".☆16Updated 4 months ago
- A Newtonian message passing network for deep learning of interatomic potentials and forces☆40Updated last month
- A framework for performing active learning for training machine-learned interatomic potentials.☆32Updated 4 months ago