Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
☆116Oct 11, 2025Updated 6 months ago
Alternatives and similar repositories for erwin
Users that are interested in erwin are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆19Nov 16, 2025Updated 5 months ago
- ☆57Jul 21, 2025Updated 9 months ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Nov 18, 2025Updated 5 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆30Feb 24, 2025Updated last year
- Implementation of the Euclidean fast attention (EFA) algorithm☆93Jan 7, 2026Updated 3 months ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- ☆12Jan 9, 2024Updated 2 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆91May 30, 2024Updated last year
- Pure C implementation of e3nn☆25Mar 17, 2025Updated last year
- Code for the paper Universal Physics Transformers☆157Jul 2, 2025Updated 10 months ago
- ☆16Jan 23, 2025Updated last year
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆303Apr 13, 2026Updated 3 weeks ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆60Sep 26, 2025Updated 7 months ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆55May 7, 2025Updated 11 months ago
- Simulation framework for a pixelated Liquid Argon TPC☆10Updated this week
- 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.
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated 2 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆120Apr 22, 2026Updated 2 weeks ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆31Feb 25, 2025Updated last year
- PaiNN in jax☆11Jan 14, 2025Updated 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…☆388Apr 22, 2026Updated 2 weeks ago
- ☆13Mar 21, 2024Updated 2 years ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆141Apr 15, 2026Updated 3 weeks ago
- Compute neighbor lists for atomistic systems☆79Apr 24, 2026Updated last week
- ☆233Feb 7, 2025Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- single-cell latent diffusion model☆49Apr 27, 2026Updated last week
- ☆70Mar 24, 2025Updated last year
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 2 years ago
- ☆22May 7, 2025Updated 11 months ago
- Inducing Point Operator Transformer: A Flexible and Scalable Architecture for Solving PDEs (AAAI 2024)☆15Jul 30, 2024Updated last year
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Apr 13, 2024Updated 2 years ago
- GINOT is a deep learning model that combines transformers with neural operators for accurate forward predictions on arbitrary 2D and 3D g…☆34Jan 20, 2026Updated 3 months ago
- Code for the paper LaM-SLidE - Latent Space Modeling of Spatial Dynamical Systems via Linked Entities☆26May 23, 2025Updated 11 months ago
- A molecular simulation package integrating MLFFs in MOFs for DAC☆47Oct 17, 2025Updated 6 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Write your code as tree-like expressions, then transform it☆21Jan 9, 2024Updated 2 years ago
- E(3) Steerable Graph Neural Network☆128Feb 10, 2023Updated 3 years ago
- ☆12May 10, 2024Updated last year
- Implementation of FlowSite and HarmonicFlow from the paper "Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding …☆106Jul 30, 2024Updated last year
- An easy (and fast) API for popular 3D molecular datasets!☆48Mar 24, 2026Updated last month
- 🌟 [NeurIPS '25 Spotlight] Fair and transparent benchmark of machine learning interatomic potentials (MLIPs), beyond basic error metrics …☆98Apr 16, 2026Updated 2 weeks ago
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆20Jan 18, 2025Updated last year