☆22May 7, 2025Updated last year
Alternatives and similar repositories for allegro-jax
Users that are interested in allegro-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆25Nov 1, 2023Updated 2 years ago
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated 5 months ago
- Equivariant machine learning interatomic potentials in JAX.☆107Apr 25, 2026Updated 3 months ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆43Mar 1, 2025Updated last year
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- An ASE-friendly implementation of the amorphous-to-crystalline (a2c) workflow.☆18Oct 19, 2025Updated 9 months ago
- [ICML'26] Phonon fine-tuning (PFT) and [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget☆76Apr 5, 2026Updated 3 months ago
- A library for building equivariant neural networks and a zoo of implementations & examples.☆31Aug 9, 2022Updated 3 years ago
- ☆18Feb 26, 2026Updated 5 months ago
- Compute neighbor lists for atomistic systems☆83Updated this week
- jax library for E3 Equivariant Neural Networks☆233Apr 1, 2026Updated 3 months ago
- Stable AutoCorrelation Integral Estimator☆17May 18, 2026Updated 2 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆30Feb 24, 2025Updated last year
- ☆11Sep 16, 2024Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆161Updated this week
- Pure C implementation of e3nn☆25Mar 17, 2025Updated last year
- ☆16Apr 22, 2025Updated last year
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated last year
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆20Nov 16, 2025Updated 8 months ago
- An overview of literature that discusses the use of machine learning for atomistic simulations☆44Apr 11, 2023Updated 3 years ago
- Force-field-enhanced Neural Networks optimized library☆88Jun 29, 2026Updated 3 weeks ago
- Training Neural Network potentials through customizable routines in JAX.☆73Jun 30, 2026Updated 3 weeks ago
- LES extension package for NequIP☆18May 26, 2026Updated 2 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆125Jul 2, 2026Updated 3 weeks ago
- CUDA implementations of MACE models☆23Aug 19, 2025Updated 11 months ago
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆74Nov 2, 2024Updated last year
- Self-describing sparse tensor data format for atomistic machine learning and beyond☆106Updated this week
- Pseudopotential converter from upf to psp8☆11Jan 25, 2023Updated 3 years ago
- Particle-mesh based calculations of long-range interactions in JAX☆30Updated this week
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆416Updated this week
- Tensor Atomic Cluster Expansion☆53Updated this week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- MESS: Modern Electronic Structure Simulations☆43May 9, 2026Updated 2 months ago
- ☆19Mar 14, 2023Updated 3 years ago
- e3nn tutorial for Materials Research Society Fall Meeting 2021☆14Nov 29, 2021Updated 4 years ago
- Allegro-pol extends the Allegro architecture to model the electric response of materials☆39May 30, 2026Updated last month
- PaiNN in jax☆11Jan 14, 2025Updated last year
- Jax / Haiku implementation of DimeNet++.☆18Mar 31, 2022Updated 4 years ago
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 2 years ago