Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0
☆20Nov 16, 2025Updated 8 months ago
Alternatives and similar repositories for e3tools
Users that are interested in e3tools are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆30Feb 24, 2025Updated last year
- An easy (and fast) API for popular 3D molecular datasets!☆51May 3, 2026Updated 2 months ago
- [ICML'25] The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs in Equivariant Tensor Products☆19Jul 16, 2025Updated last year
- Pure C implementation of e3nn☆25Mar 17, 2025Updated last year
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated 5 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.
- Time-reversal Euclidean neural networks based on e3nn☆19Sep 29, 2024Updated last year
- Bridging Smoothed Molecular Dynamics and Score-Based Learning for Conformational Ensembles☆20May 16, 2026Updated 2 months ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆101Jan 7, 2026Updated 6 months ago
- [ICML'26] Phonon fine-tuning (PFT) and [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget☆76Apr 5, 2026Updated 3 months ago
- ☆22May 7, 2025Updated last year
- High-performance CUDA kernels for equivariant graph neural networks (MACE, NequIP, Allegro). 10-20x faster than e3nn.☆44Jan 17, 2026Updated 6 months ago
- PaiNN in jax☆11Jan 14, 2025Updated last year
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆161Updated this week
- Zero Shot Molecular Generation via Similarity Kernels☆29Aug 27, 2025Updated 10 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Fully-Differentiable Tensor Spherical Harmonics in JAX☆21Apr 11, 2026Updated 3 months ago
- ☆12May 10, 2024Updated 2 years ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆31Feb 25, 2025Updated last year
- ☆27Apr 17, 2026Updated 3 months ago
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆74Nov 2, 2024Updated last year
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆61Sep 26, 2025Updated 9 months ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- jax library for E3 Equivariant Neural Networks☆233Apr 1, 2026Updated 3 months ago
- Tools for building equivariant polynomials on reductive Lie groups.☆38Jun 29, 2023Updated 3 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Script to generate distorted perovskite structures☆13Feb 10, 2024Updated 2 years ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆75May 25, 2026Updated 2 months ago
- Awesome papers related to generative flow matching and its applications in bioinformatics. (Nature Machine Intelligence)☆44Apr 24, 2026Updated 3 months ago
- ☆17Dec 14, 2025Updated 7 months ago
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆20Jan 18, 2025Updated last year
- Score-based 3D molecule generation with neural fields - NeurIPS 2024☆27Dec 3, 2024Updated last year
- Training Neural Network potentials through customizable routines in JAX.☆73Jun 30, 2026Updated 3 weeks ago
- Package of useful sampling algorithms written in MLX.☆18Feb 27, 2024Updated 2 years ago
- Official repository for the paper "Higher-Rank Irreducible Cartesian Tensors for Equivariant Message Passing".☆20Dec 4, 2025Updated 7 months 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.
- Differentiable RNA folding in JAX☆30Mar 25, 2025Updated last year
- Framework for modified sampling from biomolecular generative models☆21Updated this week
- ☆141May 7, 2024Updated 2 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆93May 30, 2024Updated 2 years ago
- ☆12Nov 22, 2024Updated last year
- MESS: Modern Electronic Structure Simulations☆21Sep 24, 2024Updated last year
- GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU☆24Mar 27, 2025Updated last year