tk-rusch / unicornn
Official code for UnICORNN (ICML 2021)
☆27Updated 3 years ago
Alternatives and similar repositories for unicornn:
Users that are interested in unicornn are comparing it to the libraries listed below
- Official code for Coupled Oscillatory RNN (ICLR 2021, Oral)☆43Updated 3 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- ☆18Updated 3 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆24Updated 3 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 4 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 3 years ago
- ☆35Updated 2 years ago
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆54Updated 3 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆16Updated 3 years ago
- ☆51Updated 6 months ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- ☆45Updated last year
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Models and code for the ICLR 2020 workshop paper "Towards Understanding Normalization in Neural ODEs"☆16Updated 4 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆24Updated 2 years ago
- ☆53Updated 6 months ago
- ☆31Updated 4 years ago
- Experiments from the paper "On Second Order Behaviour in Augmented Neural ODEs"☆57Updated 4 months ago
- Re-implementation of Hamiltonian Generative Networks paper☆33Updated 2 years ago
- Pytorch implementation of the Power Spherical distribution☆74Updated 6 months ago
- BIVA: A Very Deep Hierarchy of Latent Variables forGenerative Modeling☆29Updated 5 years ago
- ☆11Updated 4 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 5 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆13Updated 5 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆13Updated 6 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago