fabiannagel / schnax
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
- ☆21Updated last year
- ☆23Updated last year
- Training Neural Network potentials through customizable routines in JAX.☆25Updated 2 months ago
- Jax / Haiku implementation of DimeNet++.☆18Updated 3 years ago
- tools for graph-based machine-learning potentials in jax☆24Updated last year
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- Particle-mesh based calculations of long-range interactions in JAX☆17Updated 2 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆36Updated last year
- Equivariant machine learning interatomic potentials in JAX.☆72Updated last week
- ☆13Updated this week
- Nomalizing flows for orbita-free DFT☆10Updated 7 months ago
- MESS: Modern Electronic Structure Simulations☆28Updated 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
- CUDA implementations of MACE models☆14Updated 3 months ago
- ⚛ download and manipulate atomistic datasets☆44Updated 4 months ago
- Zero Shot Molecular Generation via Similarity Kernels☆19Updated last week
- ☆11Updated last month
- ☆21Updated last year
- ☆16Updated 5 months ago
- MACE_Osaka24 models☆14Updated 4 months ago
- Code for performing adversarial attacks on atomistic systems using NN potentials☆36Updated 2 years ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆46Updated last year
- ☆8Updated 2 years ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆21Updated 2 months ago
- A Newtonian message passing network for deep learning of interatomic potentials and forces☆43Updated this week
- MESS: Modern Electronic Structure Simulations☆20Updated 7 months ago
- Basis set optimization library for quantum chemistry☆34Updated last year
- Benchmarking foundation Machine Learning Potentials with Lattice Thermal Conductivity from Anharmonic Phonons☆16Updated 6 months ago
- Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics☆56Updated this week
- Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.☆30Updated 2 years ago