Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.
☆19Jan 18, 2025Updated last year
Alternatives and similar repositories for MFA
Users that are interested in MFA are comparing it to the libraries listed below
Sorting:
- ☆34Feb 12, 2025Updated last year
- [ICML'25] The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs in Equivariant Tensor Products☆18Jul 16, 2025Updated 8 months ago
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 2 years ago
- [TMLR 2024 J2C Certification] Generalizing Denoising to Non-Equilibrium Structures Improves Equivariant Force Fields☆41Feb 11, 2025Updated last year
- [ECCV'24 Oral] Chameleon: A Data-Efficient Generalist for Dense Visual Prediction in the Wild☆13Mar 13, 2025Updated last year
- Official implementation of HEGNN, a novel high-degree equivariant graph neural network proposed in the NeurIPS 2024 paper 'Are High-Degre…☆33Nov 8, 2024Updated last year
- ☆14Mar 2, 2023Updated 3 years ago
- Active Test-Time Adaptation: Theoretical Analyses and An Algorithm [ICLR 2024]☆24Nov 4, 2024Updated last year
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆69Nov 2, 2024Updated last year
- Tools for building equivariant polynomials on reductive Lie groups.☆36Jun 29, 2023Updated 2 years ago
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Oct 26, 2022Updated 3 years ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Feb 25, 2025Updated last year
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated 3 weeks ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆28Feb 24, 2025Updated last year
- A codebase dedicated to exploring multimodal learning approaches by integrating images of host galaxies of supernovae and their correspon…☆17Jan 15, 2025Updated last year
- ☆24Nov 1, 2023Updated 2 years ago
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆19Nov 16, 2025Updated 4 months ago
- A PyTorch implementation of The ICLR 2019 paper "Invariant and Equivariant Graph Networks" by Haggai Maron, Heli Ben-Hamu, Nadav Shamir a…☆17Mar 7, 2022Updated 4 years ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Nov 18, 2025Updated 4 months ago
- Pytorch implementation of "Very Deep Graph Neural Networks via Noise Regularisation"☆10Aug 22, 2021Updated 4 years ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆61Jan 7, 2026Updated 2 months ago
- ☆132May 7, 2024Updated last year
- Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"☆13Jul 2, 2023Updated 2 years ago
- JAX-based linear Einstein-Boltzmann solver for cosmology☆24Nov 13, 2025Updated 4 months ago
- ☆12Jan 9, 2024Updated 2 years ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆114Apr 7, 2025Updated 11 months ago
- E(3) Steerable Graph Neural Network☆127Feb 10, 2023Updated 3 years 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 3 months ago
- Pure C implementation of e3nn☆24Mar 17, 2025Updated last year
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆43Mar 1, 2025Updated last year
- ☆12May 10, 2024Updated last year
- ☆28Nov 4, 2024Updated last year
- do the performance test of ABACUS☆21Mar 13, 2026Updated last week
- Code for reproducing the paper "Space-Time Continuous PDE Forecasting using Equivariant Neural Fields" (https://arxiv.org/abs/2406.06660)…☆14Nov 18, 2025Updated 4 months ago
- Computational tools for simulation of high-entropy alloy surfaces☆18Jun 17, 2025Updated 9 months ago
- [ICLR 2021] Group Equivariant Generative Adversarial Networks.☆14May 6, 2021Updated 4 years ago
- An implementation of SchNet in JAX and JAX-MD.☆17Apr 5, 2022Updated 3 years ago
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆89May 30, 2024Updated last year