f-dangel / vivitLinks
[TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivatives & Newton steps
☆17Updated last year
Alternatives and similar repositories for vivit
Users that are interested in vivit are comparing it to the libraries listed below
Sorting:
- Convolutions and more as einsum for PyTorch☆17Updated last year
- ☆15Updated 4 years ago
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆22Updated 8 months ago
- Efficient Householder Transformation in PyTorch☆66Updated 4 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 6 months ago
- Source code for my PhD thesis: Backpropagation Beyond the Gradient☆20Updated 2 years ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆19Updated 3 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆26Updated 2 years ago
- Sketched matrix decompositions for PyTorch☆70Updated last week
- ☆33Updated 2 years ago
- Riemannian Optimization Using JAX☆49Updated last year
- ☆60Updated 3 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Convex potential flows☆83Updated 3 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 8 months ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆87Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆54Updated 2 months ago
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆56Updated 4 years ago
- Image augmentation library for Jax☆39Updated last year
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆43Updated last year
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆79Updated last year
- Normalizing Flows using JAX☆83Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆179Updated last month
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆71Updated 2 weeks ago
- RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation☆8Updated 4 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆75Updated 11 months ago
- ☆32Updated 9 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated 2 years ago