toshas / torch-householder
Efficient Householder Transformation in PyTorch
☆60Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for torch-householder
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆69Updated 3 months ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Convex potential flows☆78Updated 2 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows☆30Updated 3 years ago
- MintNet: Building Invertible Neural Networks with Masked Convolutions☆39Updated 3 years ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆55Updated 3 years ago
- Padé Activation Units: End-to-end Learning of Activation Functions in Deep Neural Network☆64Updated 3 years ago
- ☆31Updated 4 years ago
- We got a stew going!☆27Updated last year
- ☆40Updated 5 years ago
- ☆78Updated 3 years ago
- ☆22Updated 4 years ago
- Pytorch implementation of the Power Spherical distribution☆73Updated 3 months ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆24Updated last year
- Hessian trace estimation using PyTorch and Hutch++☆18Updated 4 years ago
- Normalizing Flows in Jax☆105Updated 4 years ago
- ☆28Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago
- ☆62Updated 9 months ago
- Spectral Tensor Train Parameterization of Deep Learning Layers☆13Updated 3 years ago
- Pytorch implementation of 'Semi-Implicit Methods for Deep Neural Networks'☆24Updated 5 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- ☆52Updated 3 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆110Updated 3 weeks ago
- Neural likelihood-free methods in PyTorch.☆39Updated 4 years ago
- ☆67Updated 2 years ago