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 3 months ago
- tools for graph-based machine-learning potentials in jax☆25Updated last year
- ☆23Updated last year
- ☆21Updated 2 weeks ago
- Equivariant machine learning interatomic potentials in JAX.☆74Updated 4 months ago
- Particle-mesh based calculations of long-range interactions in JAX☆19Updated 6 months ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆32Updated last year
- Higher order equivariant graph neural networks for 3D point clouds☆41Updated 2 years ago
- Training Neural Network potentials through customizable routines in JAX.☆43Updated 3 weeks ago
- MESS: Modern Electronic Structure Simulations☆36Updated this week
- CUDA implementations of MACE models☆15Updated this week
- ☆17Updated 8 months ago
- Auto-differentiated descriptors using Enzyme☆12Updated 4 months ago
- Jax / Haiku implementation of DimeNet++.☆18Updated 3 years ago
- A JAX-based Differentiable Density Functional Theory Framework for Materials☆32Updated last week
- ☆21Updated last year
- A flexible and performant framework for training machine learning potentials.☆27Updated this week
- Compute neighbor lists for atomistic systems☆58Updated last week
- N-Dimensional MD engine with symmetry group constraints written in C☆50Updated last year
- Pretrained model for molecular wavefunctions☆38Updated last month
- Machine Learning Interatomic Potentials with the Atomic Cluster Expansion☆59Updated last week
- A Newtonian message passing network for deep learning of interatomic potentials and forces☆44Updated 2 months ago
- Self-describing sparse tensor data format for atomistic machine learning and beyond☆79Updated 2 weeks ago
- ☆24Updated 2 weeks ago
- dataset augmentation for atomistic machine learning☆20Updated last month
- Code for performing adversarial attacks on atomistic systems using NN potentials☆39Updated 2 years ago
- ⚛ download and manipulate atomistic datasets☆46Updated 8 months ago
- Computing representations for atomistic machine learning☆73Updated 3 weeks ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆109Updated 4 months ago
- MESS: Modern Electronic Structure Simulations☆20Updated 11 months ago