PASSIONLab / OpenEquivarianceLinks
OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.
☆126Updated last week
Alternatives and similar repositories for OpenEquivariance
Users that are interested in OpenEquivariance are comparing it to the libraries listed below
Sorting:
- Pure C implementation of e3nn☆23Updated 10 months ago
- Equivariant machine learning interatomic potentials in JAX.☆83Updated this week
- ☆23Updated 2 years ago
- Build neural networks for machine learning force fields with JAX☆132Updated 8 months ago
- Higher order equivariant graph neural networks for 3D point clouds☆46Updated 2 years ago
- MESS: Modern Electronic Structure Simulations☆43Updated 4 months ago
- ☆19Updated 2 years ago
- [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget and [arXiv'26] Phonon fine-tuning (PFT)☆64Updated last week
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆110Updated 9 months ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆59Updated last month
- ☆12Updated 2 years ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆57Updated 4 months ago
- Computing representations for atomistic machine learning☆75Updated last week
- MESS: Modern Electronic Structure Simulations☆20Updated last year
- ALCHEMI Toolkit-Ops is a collection of optimized batch kernels to accelerate computational chemistry and material science workflows.☆94Updated this week
- Compute neighbor lists for atomistic systems☆73Updated this week
- [npj Comp. Mat.] Higher-order equivariant neural networks for charge density prediction in materials☆71Updated 11 months ago
- Reference implementation of "SpookyNet: Learning force fields with electronic degrees of freedom and nonlocal effects"☆85Updated 3 years ago
- ☆22Updated 9 months ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆15Updated 2 months ago
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆68Updated last year
- C++/CUDA library for SO(3) equivariant operations☆25Updated 3 months ago
- [ICLR 2025] GotenNet: Rethinking Efficient 3D Equivariant Graph Neural Networks☆60Updated 3 months ago
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆36Updated last year
- train and use graph-based ML models of potential energy surfaces☆119Updated last month
- Generate and predict molecular electron densities with Euclidean Neural Networks☆49Updated 2 years ago
- Particle-mesh based calculations of long-range interactions in JAX☆23Updated this week
- LAMMPS pair styles for NequIP and Allegro deep learning interatomic potentials☆59Updated 4 months ago
- Training Neural Network potentials through customizable routines in JAX.☆62Updated 6 months ago
- High-performance operations for neural network potentials☆99Updated 2 months ago