Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
☆111Oct 11, 2025Updated 4 months ago
Alternatives and similar repositories for erwin
Users that are interested in erwin are comparing it to the libraries listed below
Sorting:
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆18Nov 16, 2025Updated 3 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆28Feb 24, 2025Updated last year
- Implementation of the Euclidean fast attention (EFA) algorithm☆61Jan 7, 2026Updated last month
- Nomalizing flows for orbita-free DFT☆11Sep 20, 2024Updated last year
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆59Sep 26, 2025Updated 5 months ago
- Pure C implementation of e3nn☆24Mar 17, 2025Updated 11 months ago
- GINOT is a deep learning model that combines transformers with neural operators for accurate forward predictions on arbitrary 2D and 3D g…☆28Jan 20, 2026Updated last month
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆367Feb 17, 2026Updated 2 weeks ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Nov 18, 2025Updated 3 months ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆89May 30, 2024Updated last year
- JAX implementation of the NequIP neural network interatomic potential☆16Feb 24, 2026Updated last week
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Feb 25, 2025Updated last year
- An easy (and fast) API for popular 3D molecular datasets!☆43Updated this week
- ☆54Jul 21, 2025Updated 7 months ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆130Feb 26, 2026Updated last week
- Implementation of FlowSite and HarmonicFlow from the paper "Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding …☆105Jul 30, 2024Updated last year
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆54May 7, 2025Updated 9 months ago
- Write your code as tree-like expressions, then transform it☆21Jan 9, 2024Updated 2 years ago
- Graph Transformers for Large Graphs☆22Apr 26, 2024Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆141Dec 20, 2023Updated 2 years ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Apr 13, 2024Updated last year
- 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
- Simulation framework for a pixelated Liquid Argon TPC☆10Feb 24, 2026Updated last week
- ☆22May 7, 2025Updated 9 months ago
- Code for the paper LaM-SLidE - Latent Space Modeling of Spatial Dynamical Systems via Linked Entities☆26May 23, 2025Updated 9 months ago
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆293Feb 21, 2026Updated last week
- ☆221Feb 7, 2025Updated last year
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 2 years ago
- ☆14Jan 23, 2025Updated last year
- jax library for E3 Equivariant Neural Networks☆224Aug 25, 2025Updated 6 months ago
- Compute neighbor lists for atomistic systems☆74Updated this week
- ☆69Mar 24, 2025Updated 11 months ago
- Graph-based operator learning in arbitrary geometries☆34Dec 19, 2025Updated 2 months ago
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆16Nov 20, 2025Updated 3 months ago
- Implementation of the Pairformer model used in AlphaFold 3☆14Updated this week
- PaiNN in jax☆11Jan 14, 2025Updated last year
- [ICML'25] The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs in Equivariant Tensor Products☆15Jul 16, 2025Updated 7 months ago
- A Lorentz-Equivariant Transformer for All of the LHC☆29Jun 11, 2025Updated 8 months ago