fabiannagel / schnax
An implementation of SchNet in JAX and JAX-MD.
☆17Updated 2 years ago
Alternatives and similar repositories for schnax:
Users that are interested in schnax are comparing it to the libraries listed below
- ☆23Updated last year
- ☆21Updated 11 months ago
- Jax / Haiku implementation of DimeNet++.☆18Updated 2 years 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
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- Equivariant machine learning interatomic potentials in JAX.☆72Updated last year
- Higher order equivariant graph neural networks for 3D point clouds☆36Updated last year
- Training Neural Network potentials through customizable routines in JAX.☆23Updated last month
- CUDA implementations of MACE models☆14Updated 2 months ago
- MESS: Modern Electronic Structure Simulations☆27Updated 5 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
- Zero Shot Molecular Generation via Similarity Kernels☆19Updated 2 months ago
- Compute neighbor lists for atomistic systems☆49Updated 2 weeks ago
- ☆15Updated 3 months ago
- ☆21Updated last year
- Auto-differentiated descriptors using Enzyme☆12Updated last week
- ☆8Updated last year
- ⚛ download and manipulate atomistic datasets☆43Updated 3 months ago
- MESS: Modern Electronic Structure Simulations☆19Updated 6 months ago
- ☆17Updated 2 years ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆18Updated last month
- Fair and transparent benchmark of machine-learned interatomic potentials (MLIPs), beyond basic error metrics☆56Updated this week
- Machine Learning Interatomic Potentials with the Atomic Cluster Expansion☆55Updated last month
- A Newtonian message passing network for deep learning of interatomic potentials and forces☆40Updated last month
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆45Updated this week
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆46Updated last year
- Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.☆29Updated 2 years ago
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆29Updated 3 months ago
- python workflow toolkit☆37Updated last month