sophiaas / spectral-universalityLinks
Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"
☆40Updated last year
Alternatives and similar repositories for spectral-universality
Users that are interested in spectral-universality are comparing it to the libraries listed below
Sorting:
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆50Updated 6 months ago
- The Energy Transformer block, in JAX☆63Updated 2 years ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Updated 10 months ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆77Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Implementation of approximate free-energy minimization in PyTorch☆21Updated 4 years ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- Squared Non-monotonic Probabilistic Circuits☆25Updated 11 months ago
- Fast singularity detection with kernel☆37Updated 2 years ago
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 4 years ago
- Neural Optimal Transport with Lagrangian Costs☆60Updated 7 months ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆51Updated 8 months ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆45Updated 2 years ago
- About A collection of AWESOME things about information geometry Topics☆175Updated last year
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆35Updated 3 months ago
- Reference sparse coding implementations for efficient learning and inference.☆26Updated 5 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆39Updated 3 years ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆165Updated last year
- The Official PyTorch Implementation of "Brain-like Variational Inference" (NeurIPS 2025 Paper)☆64Updated last month
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆32Updated 2 years ago
- Algorithms for computations on random manifolds made easier☆94Updated 2 years ago
- This repository contains code for applying Riemannian geometry in machine learning.☆78Updated 4 years ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆46Updated 2 years ago
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Updated 3 years ago
- JAX exponential map normalising flows on sphere☆17Updated 5 years ago
- Deep Networks Grok All the Time and Here is Why☆38Updated last year
- ☆22Updated 4 years ago
- ☆52Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago