akshayka / hessian_trace_estimation
Hessian trace estimation using PyTorch and Hutch++
☆18Updated 3 years ago
Related projects: ⓘ
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated last year
- ☆49Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆34Updated 3 years ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆43Updated last year
- Inference on non-linear dynamical systems written in JAX☆12Updated 4 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Limitations of the Empirical Fisher Approximation☆45Updated 4 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- A differentiation API for PyTorch☆29Updated 4 years ago
- Official code for UnICORNN (ICML 2021)☆27Updated 2 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆39Updated 4 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆13Updated 5 years ago
- ☆31Updated 3 years ago
- Efficient Riemannian Optimization on Stiefel Manifold via Cayley Transform☆34Updated 5 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆55Updated 3 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆25Updated 2 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 5 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Updated 3 years ago
- [ICLR 2022] Path integral sampler☆40Updated last year
- This is an article about using variational autoencoders for the generation of new data. It contains the code for generating the plots and…☆12Updated 3 years ago
- Convex potential flows☆78Updated 2 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆35Updated last year
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 3 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 3 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated last year
- ☆17Updated 2 years ago
- Variational Reinforcement Learning☆16Updated last month
- Efficient Householder Transformation in PyTorch☆58Updated 3 years ago