[ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Products"
☆69Nov 2, 2024Updated last year
Alternatives and similar repositories for Gaunt-Tensor-Product
Users that are interested in Gaunt-Tensor-Product are comparing it to the libraries listed below
Sorting:
- ☆34Feb 12, 2025Updated last year
- Higher order equivariant graph neural networks for 3D point clouds☆46Jun 27, 2023Updated 2 years ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆113Apr 7, 2025Updated 10 months ago
- [ICLR 2024] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations☆323Feb 11, 2025Updated last year
- Pure C implementation of e3nn☆24Mar 17, 2025Updated 11 months ago
- ☆22May 7, 2025Updated 10 months ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆130Feb 26, 2026Updated last week
- jax library for E3 Equivariant Neural Networks☆224Aug 25, 2025Updated 6 months ago
- [TMLR 2024 J2C Certification] Generalizing Denoising to Non-Equilibrium Structures Improves Equivariant Force Fields☆40Feb 11, 2025Updated last year
- Tools for building equivariant polynomials on reductive Lie groups.☆36Jun 29, 2023Updated 2 years ago
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 2 years ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆61Jan 7, 2026Updated 2 months ago
- [npj Comp. Mat.] Higher-order equivariant neural networks for charge density prediction in materials☆71Feb 21, 2025Updated last year
- [ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs☆270Feb 11, 2025Updated last year
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆18Nov 16, 2025Updated 3 months ago
- Multi-language library for the calculation of spherical harmonics in Cartesian coordinates☆94Feb 17, 2026Updated 2 weeks ago
- Tools for geometric learning☆12Sep 26, 2025Updated 5 months ago
- dftio is to assist machine learning communities to transcript DFT output into a format that is easy to read or used by machine learning m…☆15Dec 18, 2025Updated 2 months ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆16Nov 20, 2025Updated 3 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆28Feb 24, 2025Updated last year
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆19Jan 18, 2025Updated last year
- E(3) Steerable Graph Neural Network☆126Feb 10, 2023Updated 3 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆89May 30, 2024Updated last year
- CUDA implementations of MACE models☆23Aug 19, 2025Updated 6 months ago
- Equivariant network to predict activation barriers and molecular orbitals through coefficients of molecular orbitals.☆12Feb 13, 2024Updated 2 years ago
- ☆12Oct 10, 2023Updated 2 years ago
- Code for “From Molecules to Materials Pre-training Large Generalizable Models for Atomic Property Prediction”.☆62Oct 22, 2024Updated last year
- ☆43Updated this week
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆68Dec 16, 2025Updated 2 months ago
- [ICLR 2024] "Latent 3D Graph Diffusion" by Yuning You, Ruida Zhou, Jiwoong Park, Haotian Xu, Chao Tian, Zhangyang Wang, Yang Shen☆54Jun 20, 2024Updated last year
- A library for building equivariant neural networks and a zoo of implementations & examples.☆31Aug 9, 2022Updated 3 years ago
- ☆20Dec 11, 2024Updated last year
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆368Feb 17, 2026Updated 2 weeks ago
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆39Dec 17, 2024Updated last year
- High-performance operations for neural network potentials☆102Feb 4, 2026Updated last month
- Equivariant machine learning interatomic potentials in JAX.☆87Feb 10, 2026Updated 3 weeks ago
- 🌟 [NeurIPS '25 Spotlight] Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics …☆92Feb 23, 2026Updated last week
- Collection of tutorials to use the MACE machine learning force field.☆53Jan 22, 2026Updated last month
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆53Jun 13, 2023Updated 2 years ago