Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.
☆91May 30, 2024Updated last year
Alternatives and similar repositories for ponita
Users that are interested in ponita are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆14Jan 23, 2025Updated last year
- Continuous regular group convolutions for Pytorch☆12Jun 9, 2024Updated last year
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆69Nov 2, 2024Updated last year
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Nov 18, 2025Updated 4 months ago
- E(3) Steerable Graph Neural Network☆127Feb 10, 2023Updated 3 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.
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆42Mar 1, 2025Updated last year
- ☆12Jan 9, 2024Updated 2 years ago
- Neural SPH☆43Nov 23, 2025Updated 4 months ago
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆19Nov 16, 2025Updated 4 months ago
- ☆22May 7, 2025Updated 10 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆112Oct 11, 2025Updated 5 months ago
- Paper list for equivariant neural network☆1,085Aug 19, 2024Updated last year
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆69Mar 9, 2026Updated 3 weeks ago
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 2 years ago
- 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.
- 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
- Implementation of the Euclidean fast attention (EFA) algorithm☆61Jan 7, 2026Updated 2 months ago
- E3xSO3 convolution implementation presented at MIDL 2023 https://openreview.net/pdf?id=lri_iAbpn_r☆14Apr 12, 2023Updated 2 years ago
- FLOPS counter for all your GPU benchmarking needs☆13Aug 8, 2024Updated last year
- [NeurIPS 2024] source code for "A Recipe for Charge Density Prediction"☆40Dec 17, 2024Updated last year
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆28Feb 24, 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
- ☆31Nov 23, 2025Updated 4 months ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆53Jun 13, 2023Updated 2 years 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.
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆19Jan 18, 2025Updated last year
- JAX implementation of the NequIP neural network interatomic potential☆17Feb 24, 2026Updated last month
- A modular framework for neural networks with Euclidean symmetry☆1,228Feb 13, 2026Updated last month
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆512Oct 31, 2024Updated last year
- ☆526Feb 23, 2022Updated 4 years ago
- [TMLR 2024 J2C Certification] Generalizing Denoising to Non-Equilibrium Structures Improves Equivariant Force Fields☆42Feb 11, 2025Updated last year
- Cross-platform Optimizer for ML Interatomic Potentials☆24Aug 31, 2025Updated 6 months ago
- ☆12Mar 25, 2024Updated 2 years ago
- Spatio-Spectral Graph Neural Networks (S²GNN)☆19Feb 25, 2025Updated last year
- 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.
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆142Dec 20, 2023Updated 2 years ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆380Updated this week
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Apr 13, 2024Updated last year
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- Equivariant Spherical Deconvolution: Learning Sparse Orientation Distribution Functions from Spherical Data (Pytorch)☆18Sep 8, 2022Updated 3 years ago
- TFN layers built using Tensorflow 2☆15Jul 15, 2021Updated 4 years ago
- jax library for E3 Equivariant Neural Networks☆225Aug 25, 2025Updated 7 months ago