mariogeiger / allegro-jax
☆21Updated 11 months ago
Alternatives and similar repositories for allegro-jax:
Users that are interested in allegro-jax are comparing it to the libraries listed below
- ☆23Updated last year
- An implementation of SchNet in JAX and JAX-MD.☆17Updated 2 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆36Updated last year
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- Equivariant machine learning interatomic potentials in JAX.☆72Updated last year
- Jax / Haiku implementation of DimeNet++.☆18Updated 2 years ago
- CUDA implementations of MACE models☆14Updated 2 months ago
- ☆17Updated 2 years ago
- MESS: Modern Electronic Structure Simulations☆27Updated 5 months ago
- Particle-mesh based calculations of long-range interactions in JAX☆14Updated 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
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆18Updated last month
- Zero Shot Molecular Generation via Similarity Kernels☆19Updated 2 months ago
- MESS: Modern Electronic Structure Simulations☆19Updated 6 months ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆45Updated this week
- Benchmarking foundation Machine Learning Potentials with Lattice Thermal Conductivity from Anharmonic Phonons☆15Updated 4 months ago
- Pure C implementation of e3nn☆18Updated last week
- Training Neural Network potentials through customizable routines in JAX.☆23Updated last month
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆29Updated 3 months ago
- Compute neighbor lists for atomistic systems☆49Updated 2 weeks ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆13Updated last month
- A Newtonian message passing network for deep learning of interatomic potentials and forces☆40Updated last month
- A library for building equivariant neural networks and a zoo of implementations & examples.☆32Updated 2 years ago
- ☆12Updated 10 months ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆44Updated 2 weeks ago
- tools for graph-based machine-learning potentials in jax☆20Updated 11 months ago
- Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.☆29Updated 2 years ago
- ☆12Updated last year
- Training and evaluating machine learning models for atomistic systems.☆27Updated this week
- A repository to hold installation recipes and data files for the PET-MAD universal machine-learning interatomic potential☆36Updated this week