mariogeiger / allegro-jaxView external linksLinks
☆22May 7, 2025Updated 9 months ago
Alternatives and similar repositories for allegro-jax
Users that are interested in allegro-jax are comparing it to the libraries listed below
Sorting:
- ☆23Nov 1, 2023Updated 2 years ago
- JAX implementation of the NequIP neural network interatomic potential☆14Jul 23, 2025Updated 6 months ago
- An ASE-friendly implementation of the amorphous-to-crystalline (a2c) workflow.☆18Oct 19, 2025Updated 3 months ago
- Equivariant machine learning interatomic potentials in JAX.☆86Feb 10, 2026Updated last week
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆41Mar 1, 2025Updated 11 months ago
- Compute neighbor lists for atomistic systems☆74Updated this week
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated 10 months ago
- ☆11Sep 16, 2024Updated last year
- A library for building equivariant neural networks and a zoo of implementations & examples.☆31Aug 9, 2022Updated 3 years ago
- An overview of literature that discusses the use of machine learning for atomistic simulations☆44Apr 11, 2023Updated 2 years ago
- Particle-mesh based calculations of long-range interactions in JAX☆23Feb 10, 2026Updated last week
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 2 years ago
- Pure C implementation of e3nn☆24Mar 17, 2025Updated 11 months ago
- jax library for E3 Equivariant Neural Networks☆224Aug 25, 2025Updated 5 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆28Feb 24, 2025Updated 11 months ago
- ☆18Dec 11, 2024Updated last year
- An easy (and fast) API for popular 3D molecular datasets!☆32Updated this week
- ☆42Jan 27, 2026Updated 3 weeks ago
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆18Nov 16, 2025Updated 3 months ago
- Jax / Haiku implementation of DimeNet++.☆18Mar 31, 2022Updated 3 years ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆60Jan 7, 2026Updated last month
- CUDA implementations of MACE models☆23Aug 19, 2025Updated 5 months ago
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆68Nov 2, 2024Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆110Apr 7, 2025Updated 10 months ago
- Pseudopotential converter from upf to psp8☆11Jan 25, 2023Updated 3 years ago
- Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)☆81Jan 7, 2026Updated last month
- MESS: Modern Electronic Structure Simulations☆43Sep 26, 2025Updated 4 months ago
- Self-describing sparse tensor data format for atomistic machine learning and beyond☆93Feb 6, 2026Updated last week
- ☆19Mar 14, 2023Updated 2 years ago
- Train, fine-tune, and manipulate machine learning models for atomistic systems☆54Feb 10, 2026Updated last week
- Cross-platform Optimizer for ML Interatomic Potentials☆23Aug 31, 2025Updated 5 months ago
- ☆101Nov 20, 2024Updated last year
- ☆31Jul 8, 2025Updated 7 months ago
- Training Neural Network potentials through customizable routines in JAX.☆61Jul 30, 2025Updated 6 months ago
- Force-field-enhanced Neural Networks optimized library☆81Feb 9, 2026Updated last week
- Heat-conductivity benchmark test for foundational machine-learning potentials☆29Jan 29, 2026Updated 2 weeks ago
- This is a simple but efficient implementation of PaiNN-model for constructing machine learning interatomic potentials☆25Nov 21, 2022Updated 3 years ago
- PaiNN in jax☆11Jan 14, 2025Updated last year