emilemathieu / escnn_jax
Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
☆29Updated last year
Related projects ⓘ
Alternatives and complementary repositories for escnn_jax
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆76Updated 5 months ago
- Graph neural networks in JAX.☆67Updated 5 months ago
- ☆49Updated last year
- jax library for E3 Equivariant Neural Networks☆184Updated last month
- PySCF on IPU☆42Updated 8 months ago
- Algorithms for computations on random manifolds made easier☆86Updated 11 months ago
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆21Updated this week
- Riemannian Optimization Using JAX☆45Updated last year
- A collection of graph neural networks implementations in JAX☆31Updated 11 months ago
- Normalizing Flows using JAX☆82Updated 11 months ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆41Updated 9 months ago
- Tools for building equivariant polynomials on reductive Lie groups.☆26Updated last year
- Implementation of various equivariant models in JAX☆12Updated 7 months ago
- A library for programmatically generating equivariant layers through constraint solving☆257Updated last year
- Matrix-free linear algebra in JAX.☆106Updated 2 months ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆39Updated 3 weeks ago
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆128Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆51Updated 8 months ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆26Updated 8 months ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆23Updated 7 months ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆90Updated last week
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆105Updated last week
- Riemannian Convex Potential Maps☆68Updated last year
- JAX Arrays for human consumption☆88Updated last year
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆52Updated last year
- ☆34Updated 3 years ago
- Implementation of Action Matching☆36Updated last year
- Pytorch-like dataloaders in JAX.☆59Updated last month
- Flow-matching algorithms in JAX☆77Updated 3 months ago