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:
- ☆22Updated 2 months ago
- ☆23Updated last year
- Particle-mesh based calculations of long-range interactions in JAX☆17Updated 5 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☆32Updated last year
- Equivariant machine learning interatomic potentials in JAX.☆74Updated 3 months ago
- Training Neural Network potentials through customizable routines in JAX.☆41Updated this week
- MESS: Modern Electronic Structure Simulations☆32Updated this week
- Jax / Haiku implementation of DimeNet++.☆18Updated 3 years ago
- Auto-differentiated descriptors using Enzyme☆12Updated 4 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆40Updated 2 years ago
- ☆21Updated 2 months ago
- N-Dimensional MD engine with symmetry group constraints written in C☆49Updated last year
- Zero Shot Molecular Generation via Similarity Kernels☆23Updated 2 months ago
- ☆8Updated 2 years ago
- Infrastructure to implement and train NNPs☆18Updated last week
- CUDA implementations of MACE models☆15Updated 6 months ago
- ☆16Updated 8 months ago
- Robust NN MD simulator☆20Updated 2 years ago
- A Newtonian message passing network for deep learning of interatomic potentials and forces☆44Updated last month
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆109Updated 3 months ago
- Compute neighbor lists for atomistic systems☆57Updated this week
- ☆11Updated 5 months ago
- Pretrained model for molecular wavefunctions☆35Updated last week
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆14Updated 5 months ago
- MESS: Modern Electronic Structure Simulations☆20Updated 10 months ago
- Nomalizing flows for orbita-free DFT☆10Updated 10 months ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆50Updated 2 years ago
- A JAX-based Differentiable Density Functional Theory Framework for Materials☆30Updated last week
- Machine-Learned Interatomic Potential eXploration (mlipx) is designed at BASF for evaluating machine-learned interatomic potentials (MLIP…☆89Updated last month