nickjbrowning / cuda_mace
CUDA implementations of MACE models
☆13Updated last month
Alternatives and similar repositories for cuda_mace:
Users that are interested in cuda_mace are comparing it to the libraries listed below
- MESS: Modern Electronic Structure Simulations☆26Updated 5 months ago
- ☆21Updated 10 months ago
- ☆22Updated last year
- ☆17Updated last year
- Official repository for the paper "Uncertainty-biased molecular dynamics for learning uniformly accurate interatomic potentials".☆16Updated 4 months ago
- Equivariant machine learning interatomic potentials in JAX.☆70Updated last year
- An implementation of SchNet in JAX and JAX-MD.☆17Updated 2 years ago
- MESS: Modern Electronic Structure Simulations☆19Updated 5 months ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- Benchmarking foundation Machine Learning Potentials with Lattice Thermal Conductivity from Anharmonic Phonons☆15Updated 4 months ago
- Training Neural Network potentials through customizable routines in JAX.☆23Updated 2 weeks ago
- Compute neighbor lists for atomistic systems☆39Updated 2 weeks ago
- This repo contains the datasets used for the paper The Design Space of E(3)-Equivariant Atom-Centered Interatomic Potentials.☆18Updated 2 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆35Updated last year
- ☆20Updated 11 months ago
- Collection of Tutorials on Machine Learning Interatomic Potentials☆18Updated 7 months ago
- Particle-mesh based calculations of long-range interactions in PyTorch☆33Updated last week
- Computing representations for atomistic machine learning☆66Updated this week
- Pure C implementation of e3nn☆17Updated 3 months ago
- python workflow toolkit☆37Updated last week
- Zero Shot Molecular Generation via Similarity Kernels☆17Updated last month
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆30Updated 3 weeks ago
- Predict the electronic structure and atomic properties (potential energy, forces, and stress tensor) of polymers containing N and/or O.☆17Updated 6 months ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆13Updated this week
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆24Updated 2 months ago
- tools for graph-based machine-learning potentials in jax☆19Updated 10 months ago
- Training and evaluating machine learning models for atomistic systems.☆25Updated this week
- Particle-mesh based calculations of long-range interactions in JAX☆13Updated 2 weeks ago
- Fair and transparent benchmark of machine-learned interatomic potentials (MLIPs), beyond basic error metrics☆56Updated 2 weeks ago