nickjbrowning / cuda_mace
CUDA implementations of MACE models
☆10Updated 2 months ago
Related projects: ⓘ
- ☆17Updated 10 months ago
- ☆17Updated 5 months ago
- Equivariant machine learning interatomic potentials in JAX.☆59Updated 11 months ago
- ☆14Updated last year
- ☆10Updated 4 months ago
- An implementation of SchNet in JAX and JAX-MD.☆16Updated 2 years ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆29Updated 6 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆33Updated last year
- Compute neighbor lists for atomistic systems☆25Updated this week
- C++/CUDA library for SO(3) equivariant operations☆19Updated last week
- Collection of Tutorials on Machine Learning Interatomic Potentials☆11Updated last month
- MESS: Modern Electronic Structure Simulations☆17Updated 3 months ago
- Build neural networks for machine learning force fields with JAX☆75Updated 3 weeks ago
- Jax / Haiku implementation of DimeNet++.☆16Updated 2 years ago
- ☆12Updated 3 months ago
- Tools for building equivariant polynomials on reductive Lie groups.☆26Updated last year
- This repo contains the datasets used for the paper The Design Space of E(3)-Equivariant Atom-Centered Interatomic Potentials.☆17Updated 2 years ago
- ☆41Updated 3 months ago
- Computing representations for atomistic machine learning☆44Updated this week
- MACE-OFF23 models☆19Updated 5 months ago
- Official repository for the paper "Uncertainty-biased molecular dynamics for learning uniformly accurate interatomic potentials".☆9Updated 6 months ago
- ☆11Updated 11 months ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆41Updated last year
- ☆39Updated 3 weeks ago
- ☆17Updated 5 months ago
- DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electr…☆47Updated 3 months ago
- Generative materials benchmarking metrics, inspired by guacamol and CDVAE.☆34Updated 3 months ago
- ☆13Updated 6 months ago
- python workflow toolkit☆31Updated 2 weeks ago
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆17Updated 3 months ago