nickjbrowning / cuda_mace
CUDA implementations of MACE models
☆14Updated 3 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 last year
- ☆23Updated last year
- ☆18Updated 2 years ago
- MESS: Modern Electronic Structure Simulations☆28Updated 3 weeks ago
- Benchmarking foundation Machine Learning Potentials with Lattice Thermal Conductivity from Anharmonic Phonons☆16Updated 5 months ago
- MESS: Modern Electronic Structure Simulations☆20Updated 7 months ago
- Official repository for the paper "Uncertainty-biased molecular dynamics for learning uniformly accurate interatomic potentials".☆20Updated 5 months ago
- Equivariant machine learning interatomic potentials in JAX.☆72Updated this week
- Compute neighbor lists for atomistic systems☆52Updated 2 weeks ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- An implementation of SchNet in JAX and JAX-MD.☆17Updated 3 years ago
- tools for graph-based machine-learning potentials in jax☆23Updated last year
- Particle-mesh based calculations of long-range interactions in JAX☆17Updated 2 months ago
- Pure C implementation of e3nn☆18Updated 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
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆49Updated this week
- C++/CUDA library for SO(3) equivariant operations☆23Updated this week
- ☆21Updated last year
- Training Neural Network potentials through customizable routines in JAX.☆24Updated 2 months ago
- Collection of Tutorials on Machine Learning Interatomic Potentials☆18Updated 9 months ago
- Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics☆56Updated last week
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆32Updated last month
- Zero Shot Molecular Generation via Similarity Kernels☆19Updated 3 months ago
- ☆12Updated 11 months ago
- Nomalizing flows for orbita-free DFT☆10Updated 7 months ago
- Steerable E(3) GNN in jax☆24Updated last year
- python workflow toolkit☆38Updated 2 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆36Updated last year
- Collection of tutorials to use the MACE machine learning force field.☆45Updated 7 months ago
- Input files for Batzner, S., Musaelian, A., Sun, L., Geiger, M., Mailoa, J. P., Kornbluth, M., ... & Kozinsky, B. (2021). E(3)-equivarian…☆13Updated 3 years ago