IntelLabs / EquiTriton
EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks, enabling compute efficient training and inference.
☆53Updated this week
Alternatives and similar repositories for EquiTriton:
Users that are interested in EquiTriton are comparing it to the libraries listed below
- PySCF on IPU☆42Updated 10 months ago
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆58Updated 2 months ago
- Pure C implementation of e3nn☆16Updated 2 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…☆163Updated this week
- Personal solutions to the Triton Puzzles☆18Updated 6 months ago
- Collection of kernels written in Triton language☆90Updated 2 months ago
- Sparsity support for PyTorch☆33Updated last month
- A lightweight ab initio molecular dynamics simulation program☆39Updated 9 months ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆32Updated last month
- Experiment of using Tangent to autodiff triton☆74Updated 11 months ago
- This Denoising Force Field (DFF) codebase provides a Pytorch framework for the method presented in Two for one: Diffusion models and forc…☆53Updated 7 months ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆45Updated last month
- ☆74Updated last year
- This is a port of Mistral-7B model in JAX☆30Updated 6 months ago
- [TMLR 2024] Generalizing Denoising to Non-Equilibrium Structures Improves Equivariant Force Fields☆29Updated 3 weeks ago
- Code for “FlowMM Generating Materials with Riemannian Flow Matching” and "FlowLLM: Flow Matching for Material Generation with Large Langu…☆105Updated 2 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆16Updated this week
- C++17 implementation of einops for libtorch - clear and reliable tensor manipulations with einstein-like notation☆10Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆96Updated 2 months ago
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆23Updated last month
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆79Updated 7 months ago
- Implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch☆53Updated last week
- The OGB-LSC is the Large Scale Competition by Open Graph Benchmark to help accelerate research into machine learning on graph structured …☆71Updated 5 months ago
- Minimal AlphaFold☆54Updated 4 months ago
- jax library for E3 Equivariant Neural Networks☆194Updated last month
- Exchange correlation functionals translated from libxc to jax☆43Updated last year
- extensible collectives library in triton☆76Updated 3 months ago
- ICML2024: Equivariant Graph Neural Operator for Modeling 3D Dynamics☆51Updated 9 months ago
- Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…☆254Updated last month
- JAX bindings for Flash Attention v2☆83Updated 6 months ago