sophiaas / spectral-universality
Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"
☆32Updated 9 months ago
Related projects ⓘ
Alternatives and complementary repositories for spectral-universality
- Neural Optimal Transport with Lagrangian Costs☆49Updated 4 months ago
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆24Updated 6 months ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year
- Differentiable Euler Characteristic Transform☆17Updated 5 months ago
- Quantify geometric intelligence in natural and artificial brains.☆48Updated 2 months ago
- ☆15Updated last year
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆18Updated 2 years ago
- The Energy Transformer block, in JAX☆53Updated 11 months ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆26Updated 8 months ago
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 3 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- Official repository of "Spontaneous symmetry breaking in generative diffusion models"☆40Updated 5 months ago
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆40Updated 3 months ago
- Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"☆14Updated this week
- Geometric Dynamic Variational Autoencoders (GD-VAEs) for learning embedding maps for nonlinear dynamics into general latent spaces. This …☆24Updated last year
- ☆34Updated 3 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆41Updated last year
- Official implementation of Deep Momentum Schrödinger Bridge☆20Updated last year
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆40Updated last year
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Updated 2 years ago
- ☆31Updated 5 months ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆70Updated 3 months ago
- ☆16Updated 11 months ago
- Representation Learning on Topological Domains☆78Updated this week
- ☆23Updated last year
- ☆20Updated last month
- Flow-matching algorithms in JAX☆77Updated 3 months ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆36Updated 5 months ago
- Exact OU processes with JAX☆34Updated last month
- ☆10Updated last year