Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
☆119Oct 11, 2025Updated 11 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☆22Nov 16, 2025Updated 10 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆30Feb 24, 2025Updated last year
- Compute neighbor lists for atomistic systems☆85Sep 10, 2026Updated 2 weeks ago
- Bridging Smoothed Molecular Dynamics and Score-Based Learning for Conformational Ensembles☆21May 16, 2026Updated 4 months ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆105Jan 7, 2026Updated 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. Zero configuration with optimized deployments.
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆44Mar 1, 2025Updated last year
- ☆63Jul 21, 2025Updated last year
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆94May 30, 2024Updated 2 years ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆60Sep 26, 2025Updated 11 months ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆58May 7, 2025Updated last year
- Fully-Differentiable Tensor Spherical Harmonics in JAX☆22Apr 11, 2026Updated 5 months ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆435Updated this week
- Official implementation of All Atom Diffusion Transformers (ICML 2025)☆318Apr 13, 2026Updated 5 months ago
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 2 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆169Updated this week
- EquiformerV3: Scaling Efficient, Expressive, and General SE(3)-Equivariant Graph Attention Transformers☆121Apr 17, 2026Updated 5 months ago
- ⚡ Triton implementation of Clifford algebra neural networks.☆42May 2, 2026Updated 4 months ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆26Nov 18, 2025Updated 10 months ago
- ☆12Jan 9, 2024Updated 2 years ago
- Code for the paper Universal Physics Transformers☆173Jul 2, 2025Updated last year
- ☆22May 7, 2025Updated last year
- Benchmarking foundation Machine Learning Potentials with Lattice Thermal Conductivity from Anharmonic Phonons☆15Oct 30, 2024Updated last year
- Implementation of FlowSite and HarmonicFlow from the paper "Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding …☆107Jul 30, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Nomalizing flows for orbita-free DFT☆12Sep 20, 2024Updated 2 years ago
- ☆12Oct 29, 2024Updated last year
- [ECCV'22] Equivariant Hypergraph Neural Networks, in PyTorch☆31Apr 11, 2023Updated 3 years ago
- Pure C implementation of e3nn☆26Mar 17, 2025Updated last year
- [ICML'26] Phonon fine-tuning (PFT) and [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget☆76Apr 5, 2026Updated 5 months ago
- ☆52Aug 19, 2026Updated last month
- [ICML 2026 GenBio Workshop] Official Implementation for "Harmonic Torsional Diffusion for Protein-Ligand Flexible Docking"☆17Jun 30, 2026Updated 2 months ago
- jax library for E3 Equivariant Neural Networks☆234Apr 1, 2026Updated 5 months ago
- Simulation framework for a pixelated Liquid Argon TPC☆11Updated this week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- An easy (and fast) API for popular 3D molecular datasets!☆50May 3, 2026Updated 4 months ago
- ORB forcefield models from Orbital Materials☆616Sep 8, 2026Updated 2 weeks ago
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated 6 months ago
- SubPEx (Sub-Pocket Explorer) is a tool to enhance ensemble (multiple-receptor/relaxed-complex) virtual screening. It uses weighted ensemb…☆18Jun 30, 2025Updated last year
- A high-performance toolkit for atomistic simulations in JAX.☆234Updated this week
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆31Feb 25, 2025Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆126Sep 9, 2026Updated 2 weeks ago