fabiannagel / schnaxLinks
An implementation of SchNet in JAX and JAX-MD.
☆17Updated 3 years ago
Alternatives and similar repositories for schnax
Users that are interested in schnax are comparing it to the libraries listed below
Sorting:
- ☆23Updated last year
- ☆22Updated last month
- Jax / Haiku implementation of DimeNet++.☆18Updated 3 years ago
- Particle-mesh based calculations of long-range interactions in JAX☆17Updated 4 months ago
- tools for graph-based machine-learning potentials in jax☆25Updated last year
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- Training Neural Network potentials through customizable routines in JAX.☆34Updated this week
- Equivariant machine learning interatomic potentials in JAX.☆73Updated last month
- MESS: Modern Electronic Structure Simulations☆31Updated 2 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆39Updated last year
- CUDA implementations of MACE models☆15Updated 5 months ago
- ☆8Updated 2 years ago
- ☆16Updated 3 weeks ago
- Auto-differentiated descriptors using Enzyme☆12Updated 2 months 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
- ⚛ download and manipulate atomistic datasets☆44Updated 6 months ago
- ☆18Updated this week
- MESS: Modern Electronic Structure Simulations☆20Updated 8 months ago
- Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.☆30Updated 3 years ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆49Updated 2 years ago
- Differentiable molecular simulation of proteins with a coarse-grained potential☆55Updated 2 months ago
- Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics https://openreview.net/pdf?…☆58Updated this week
- Compute neighbor lists for atomistic systems☆53Updated this week
- ☆21Updated last year
- Nomalizing flows for orbita-free DFT☆10Updated 9 months ago
- Code for performing adversarial attacks on atomistic systems using NN potentials☆38Updated 2 years ago
- Zero Shot Molecular Generation via Similarity Kernels☆20Updated 3 weeks ago
- ☆18Updated last month
- ☆16Updated 6 months ago
- A library for building equivariant neural networks and a zoo of implementations & examples.☆32Updated 2 years ago