nickjbrowning / cuda_mace
CUDA implementations of MACE models
☆14Updated 2 months ago
Alternatives and similar repositories for cuda_mace:
Users that are interested in cuda_mace are comparing it to the libraries listed below
- ☆21Updated 11 months ago
- ☆17Updated 2 years ago
- ☆23Updated last year
- MESS: Modern Electronic Structure Simulations☆27Updated this week
- Official repository for the paper "Uncertainty-biased molecular dynamics for learning uniformly accurate interatomic potentials".☆16Updated 5 months ago
- Equivariant machine learning interatomic potentials in JAX.☆72Updated last year
- MESS: Modern Electronic Structure Simulations☆19Updated 6 months ago
- An implementation of SchNet in JAX and JAX-MD.☆17Updated 2 years ago
- This repo contains the datasets used for the paper The Design Space of E(3)-Equivariant Atom-Centered Interatomic Potentials.☆19Updated 2 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆36Updated last year
- Benchmarking foundation Machine Learning Potentials with Lattice Thermal Conductivity from Anharmonic Phonons☆16Updated 5 months ago
- Compute neighbor lists for atomistic systems☆49Updated 3 weeks ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- ☆21Updated last year
- Pure C implementation of e3nn☆18Updated 2 weeks ago
- Training Neural Network potentials through customizable routines in JAX.☆23Updated last month
- Collection of Tutorials on Machine Learning Interatomic Potentials☆18Updated 8 months ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆46Updated last week
- C++/CUDA library for SO(3) equivariant operations☆23Updated this week
- ☆16Updated 4 months ago
- Computing representations for atomistic machine learning☆67Updated last month
- Steerable E(3) GNN in jax☆24Updated last year
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆45Updated 3 weeks ago
- Display and Edit Molecules (https://zndraw.icp.uni-stuttgart.de)☆40Updated last week
- Fair and transparent benchmark of machine-learned interatomic potentials (MLIPs), beyond basic error metrics☆56Updated 2 weeks ago
- tools for graph-based machine-learning potentials in jax☆20Updated 11 months ago
- Particle-mesh based calculations of long-range interactions in JAX☆14Updated last month
- python workflow toolkit☆37Updated last month
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆13Updated last month
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆30Updated last month