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
- ☆21Updated 9 months ago
- ☆21Updated last year
- Higher order equivariant graph neural networks for 3D point clouds☆33Updated last year
- tools for graph-based machine-learning potentials in jax☆18Updated 9 months ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated 10 months ago
- Training Neural Network potentials through customizable routines in JAX.☆22Updated last week
- Compute neighbor lists for atomistic systems☆33Updated last month
- Particle-mesh based calculations of long-range interactions in JAX☆12Updated last month
- Equivariant machine learning interatomic potentials in JAX.☆66Updated last year
- MESS: Modern Electronic Structure Simulations☆24Updated 3 months ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆32Updated last month
- Jax / Haiku implementation of DimeNet++.☆18Updated 2 years ago
- ☆15Updated 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
- A Newtonian message passing network for deep learning of interatomic potentials and forces☆38Updated 5 months ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆43Updated last year
- Zero Shot Molecular Generation via Similarity Kernels☆17Updated last month
- ☆19Updated 10 months ago
- MACE_Osaka24 models☆12Updated last month
- Collection of Tutorials on Machine Learning Interatomic Potentials☆14Updated 5 months ago
- Benchmarking foundation Machine Learning Potentials with Lattice Thermal Conductivity from Anharmonic Phonons☆14Updated 2 months ago
- Particle-mesh based calculations of long-range interactions in PyTorch☆30Updated this week
- Fair and transparent benchmark of machine-learned interatomic potentials (MLIPs), beyond basic error metrics☆55Updated this week
- ⚛ download and manipulate atomistic datasets☆39Updated last month
- Auto-differentiated descriptors using Enzyme☆12Updated 3 months ago
- Moment Invariants Local Atomic Descriptor☆31Updated 4 months ago
- Training and evaluating machine learning models for atomistic systems.☆22Updated this week
- Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.☆28Updated 2 years ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆16Updated this week
- Repository of Quantum Datasets Publicly Available☆41Updated 2 months ago