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:
- Particle-mesh based calculations of long-range interactions in JAX☆19Updated 3 weeks ago
- ☆22Updated 8 months ago
- tools for graph-based machine-learning potentials in jax☆26Updated last year
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆36Updated last year
- ☆23Updated 2 years ago
- ☆33Updated 3 weeks ago
- Equivariant machine learning interatomic potentials in JAX.☆81Updated this week
- MESS: Modern Electronic Structure Simulations☆43Updated 3 months ago
- Training Neural Network potentials through customizable routines in JAX.☆59Updated 5 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆46Updated 2 years ago
- A flexible and performant framework for training machine learning potentials.☆30Updated 2 weeks ago
- Jax / Haiku implementation of DimeNet++.☆18Updated 3 years ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆15Updated last month
- ☆10Updated 2 years ago
- Auto-differentiated descriptors using Enzyme☆12Updated 9 months ago
- A JAX-based Differentiable Density Functional Theory Framework for Materials☆42Updated last week
- MESS: Modern Electronic Structure Simulations☆20Updated last year
- ☆21Updated last year
- Compute neighbor lists for atomistic systems☆70Updated this week
- ⚛ download and manipulate atomistic datasets☆48Updated last month
- Code for performing adversarial attacks on atomistic systems using NN potentials☆40Updated 3 years ago
- ☆11Updated last year
- N-Dimensional MD engine with symmetry group constraints written in C☆50Updated last year
- CUDA implementations of MACE models☆20Updated 4 months ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆40Updated 10 months ago
- Zero Shot Molecular Generation via Similarity Kernels☆28Updated 4 months ago
- ☆12Updated 10 months ago
- Basis set optimization library for quantum chemistry☆36Updated 6 months ago
- ☆17Updated 3 months ago
- ☆14Updated last year