f-dangel / einconv
Convolutions and more as einsum for PyTorch
☆13Updated 5 months ago
Related projects ⓘ
Alternatives and complementary repositories for einconv
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- ☆15Updated 3 years ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆18Updated 3 years ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆23Updated 2 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Updated 3 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆24Updated last year
- A differentiation API for PyTorch☆30Updated 4 years ago
- ☆33Updated last year
- ☆16Updated last year
- Repository for the PopulAtion Parameter Averaging (PAPA) paper☆26Updated 6 months ago
- ☆36Updated 2 years ago
- Efficient Neural Network Loss Landscape Generation☆10Updated 5 years ago
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 3 years ago
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆25Updated last year
- Implementation of Action Matching for the Schrödinger equation☆22Updated last year
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆41Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆26Updated 3 years ago
- ☆15Updated last year
- Code for "Journey to the BAOAB-limit: finding effective MCMC samplers for score-based models". See more at https://ajayj.com/journey.☆12Updated last year
- Efficient Householder Transformation in PyTorch☆60Updated 3 years ago
- "Stochasticity in Neural ODEs: An Empirical Study". Experiments from the paper☆13Updated 4 years ago
- Hessian trace estimation using PyTorch and Hutch++☆18Updated 4 years ago
- Convex potential flows☆78Updated 2 years ago
- MintNet: Building Invertible Neural Networks with Masked Convolutions☆39Updated 3 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 3 years ago
- Probabilistic Solution of Differential Equations☆14Updated 2 years ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆35Updated 4 months ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- Source code for my PhD thesis: Backpropagation Beyond the Gradient☆20Updated last year
- ☆17Updated 2 years ago