katalinic / sdflows
JAX exponential map normalising flows on sphere
☆17Updated 4 years ago
Alternatives and similar repositories for sdflows:
Users that are interested in sdflows are comparing it to the libraries listed below
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆35Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows☆31Updated 3 years ago
- ☆33Updated last year
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated 9 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Transformers with doubly stochastic attention☆44Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆53Updated 10 months ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- Stochastic Normalizing Flows☆73Updated 3 years ago
- Convex potential flows☆81Updated 3 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆24Updated 2 years ago
- ☆16Updated 2 years ago
- ☆15Updated 4 years ago
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆72Updated 5 months ago
- ☆17Updated last year
- Efficient Householder Transformation in PyTorch☆63Updated 3 years ago
- [ICLR 2022] Path integral sampler☆43Updated last year
- PyTorch implementation of the OT-Flow approach in arXiv:2006.00104☆50Updated 5 months ago
- Experiments for the AISTATS publication on Reparameterizing Distributions over Lie Groups☆52Updated 2 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 4 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- Multiplicative Normalizing Flows in PyTorch.☆23Updated last week
- ☆31Updated 4 years ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆39Updated 7 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆38Updated 4 years ago