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☆23Updated last week
- ☆23Updated 2 years ago
- tools for graph-based machine-learning potentials in jax☆26Updated last year
- ☆22Updated 8 months ago
- Equivariant machine learning interatomic potentials in JAX.☆83Updated this week
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆36Updated last year
- ☆33Updated this week
- Higher order equivariant graph neural networks for 3D point clouds☆46Updated 2 years ago
- MESS: Modern Electronic Structure Simulations☆43Updated 4 months ago
- [ICML 2025] Repurposing pre-trained score-based generative models for transition path sampling by minimizing the Onsager-Machlup (OM) act…☆25Updated 2 weeks ago
- Jax / Haiku implementation of DimeNet++.☆18Updated 3 years ago
- A flexible and performant framework for training machine learning potentials.☆32Updated this week
- Training Neural Network potentials through customizable routines in JAX.☆60Updated 6 months ago
- Auto-differentiated descriptors using Enzyme☆12Updated 9 months ago
- N-Dimensional MD engine with symmetry group constraints written in C☆49Updated last year
- Compute neighbor lists for atomistic systems☆72Updated this week
- A JAX-based Differentiable Density Functional Theory Framework for Materials☆42Updated this week
- Zero Shot Molecular Generation via Similarity Kernels☆28Updated 5 months ago
- MESS: Modern Electronic Structure Simulations☆20Updated last year
- Computing representations for atomistic machine learning☆75Updated last week
- [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget and [arXiv'26] Phonon fine-tuning (PFT)☆64Updated this week
- ⚛ download and manipulate atomistic datasets☆48Updated 2 months ago
- ☆10Updated 2 years ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆110Updated 9 months ago
- CUDA implementations of MACE models☆20Updated 5 months ago
- Robust NN MD simulator☆21Updated 2 years ago
- Pretrained model for molecular wavefunctions☆55Updated 2 weeks ago
- Computing the Committor with the Committor: an Anatomy of the Transition State Ensemble☆12Updated last year
- Moment Invariants Local Atomic Descriptor☆34Updated last year
- ☆17Updated 4 months ago