Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
☆112Oct 11, 2025Updated 5 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 4 months ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆43Mar 1, 2025Updated last year
- Implementation of the Euclidean fast attention (EFA) algorithm☆61Jan 7, 2026Updated 2 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆28Feb 24, 2025Updated last year
- ☆56Jul 21, 2025Updated 8 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Nov 18, 2025Updated 4 months ago
- ☆12Jan 9, 2024Updated 2 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆91May 30, 2024Updated last year
- Nomalizing flows for orbita-free DFT☆11Sep 20, 2024Updated last year
- Code for the paper Universal Physics Transformers☆152Jul 2, 2025Updated 8 months ago
- Pure C implementation of e3nn☆24Mar 17, 2025Updated last year
- ☆14Jan 23, 2025Updated last year
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆296Mar 13, 2026Updated last week
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆60Sep 26, 2025Updated 6 months ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆54May 7, 2025Updated 10 months ago
- Simulation framework for a pixelated Liquid Argon TPC☆10Mar 15, 2026Updated last week
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated last month
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆115Apr 7, 2025Updated 11 months ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Feb 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…☆377Mar 16, 2026Updated last week
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆133Mar 6, 2026Updated 2 weeks ago
- single-cell latent diffusion model☆44Mar 16, 2026Updated last week
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Compute neighbor lists for atomistic systems☆74Mar 10, 2026Updated 2 weeks ago
- ☆226Feb 7, 2025Updated last year
- Tabular data imputation and generation, with flexible modeling of quantitative features via hierarchical binning (TMLR, 2025)☆16Mar 10, 2025Updated last year
- 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
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- ☆70Mar 24, 2025Updated last year
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 2 years ago
- ☆22May 7, 2025Updated 10 months ago
- Inducing Point Operator Transformer: A Flexible and Scalable Architecture for Solving PDEs (AAAI 2024)☆15Jul 30, 2024Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Apr 13, 2024Updated last year
- GINOT is a deep learning model that combines transformers with neural operators for accurate forward predictions on arbitrary 2D and 3D g…☆30Jan 20, 2026Updated 2 months ago
- Code for the paper LaM-SLidE - Latent Space Modeling of Spatial Dynamical Systems via Linked Entities☆26May 23, 2025Updated 10 months ago
- A molecular simulation package integrating MLFFs in MOFs for DAC☆43Oct 17, 2025Updated 5 months ago
- Write your code as tree-like expressions, then transform it☆21Jan 9, 2024Updated 2 years ago
- E(3) Steerable Graph Neural Network☆127Feb 10, 2023Updated 3 years ago
- ☆12May 10, 2024Updated last year