sophiaas / spectral-universality
Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"
☆38Updated last year
Alternatives and similar repositories for spectral-universality:
Users that are interested in spectral-universality are comparing it to the libraries listed below
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated 3 weeks ago
- Neural Optimal Transport with Lagrangian Costs☆55Updated 8 months ago
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Updated 2 years ago
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆47Updated 7 months ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆41Updated 2 months ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆70Updated last year
- The Energy Transformer block, in JAX☆56Updated last year
- Deep Networks Grok All the Time and Here is Why☆31Updated 10 months ago
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆26Updated 10 months ago
- Quantify geometric intelligence in natural and artificial brains.☆50Updated 3 months ago
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- Differentiable Euler Characteristic Transform☆17Updated 9 months ago
- Squared Non-monotonic Probabilistic Circuits☆22Updated 2 months ago
- Implicit^2: Implicit model for implicit neural representations☆28Updated 3 years ago
- Adjoint-Equivariant Neural Network for Semi-Simple Lie Algebras☆26Updated 3 months ago
- Exact OU processes with JAX☆42Updated this week
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Official repository of "Spontaneous symmetry breaking in generative diffusion models"☆42Updated 10 months ago
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆27Updated 2 years ago
- ☆17Updated last year
- Flexible Inference for Predictive Coding Networks in JAX.☆31Updated 2 weeks ago
- ☆24Updated last week
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆20Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- LieTransformer: Equivariant Self-Attention for Lie Groups☆69Updated 7 months ago
- On efficient computation in active inference☆16Updated 9 months ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆28Updated last year
- JAX exponential map normalising flows on sphere☆17Updated 4 years ago