david-knigge / enf-min-jax
A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).
☆13Updated 3 weeks ago
Related projects: ⓘ
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆39Updated last month
- ☆16Updated 6 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆26Updated last year
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆68Updated 3 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆46Updated last year
- ☆54Updated 3 months ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆68Updated 3 months ago
- Graph neural networks in JAX.☆66Updated 3 months ago
- Implementation of papers in 101 lines of code.☆18Updated 10 months ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆23Updated 6 months ago
- A collection of graph neural networks implementations in JAX☆30Updated 9 months ago
- Official implementation of Deep Momentum Schrödinger Bridge☆19Updated 10 months ago
- Official PyTorch implementation of the Vectorized Conditional Neural Field.☆11Updated last month
- Neural Optimal Transport with Lagrangian Costs☆37Updated 2 months ago
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆17Updated 3 months ago
- Implementation of Action Matching for the Schrödinger equation☆22Updated last year
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆34Updated 3 months ago
- ☆62Updated last year
- Lie Algebra Convolutional Network implementation☆42Updated 2 years ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆21Updated last month
- Flow-matching algorithms in JAX☆62Updated last month
- Code for Lie Symmetries SSL paper☆18Updated 6 months ago
- Code for the book "The Elements of Differentiable Programming".☆57Updated last month
- ☆49Updated last year
- Adjoint-Equivariant Neural Network for Semi-Simple Lie Algebras☆15Updated last month
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆55Updated 2 months ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆48Updated 3 weeks ago
- ☆33Updated 2 years ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆35Updated last year
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆22Updated 5 months ago