JAX exponential map normalising flows on sphere
☆17Oct 4, 2020Updated 5 years ago
Alternatives and similar repositories for sdflows
Users that are interested in sdflows are comparing it to the libraries listed below
Sorting:
- This is a Pytorch implementation of [normalizing flows on tori and spheres, ICML 2020]☆19Oct 12, 2022Updated 3 years ago
- Riemannian Convex Potential Maps☆67Mar 31, 2023Updated 2 years ago
- Quasi-Newton Algorithm for Stochastic Optimization☆11May 20, 2022Updated 3 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Oct 6, 2022Updated 3 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Jun 12, 2024Updated last year
- Fast Laplacian estimation☆16May 31, 2024Updated last year
- ☆13Mar 26, 2019Updated 6 years ago
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆126Jul 6, 2023Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- ☆14Aug 29, 2023Updated 2 years ago
- Source code and input files associated to the paper "Targeted free energy perturbation revisited: Accurate free energies from mapped refe…☆13Sep 14, 2021Updated 4 years ago
- Blog post☆17Feb 16, 2024Updated 2 years ago
- Python implementation of paper "AntisymmetricRNN: A Dynamical System View on Recurrent Neural Networks"☆15Aug 2, 2019Updated 6 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Dec 8, 2022Updated 3 years ago
- Efficient Video Prediction via Sparsely Conditioned Flow Matching. In ICCV, 2023.☆23Jun 5, 2024Updated last year
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 4 years ago
- ☆23Aug 21, 2025Updated 6 months ago
- ☆18Mar 16, 2022Updated 3 years ago
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 5 years ago
- Xmixers: A collection of SOTA efficient token/channel mixers☆28Sep 4, 2025Updated 5 months ago
- Implementation of REBAR in PyTorch☆17Jul 18, 2018Updated 7 years ago
- ☆24Oct 21, 2024Updated last year
- The Official Implementation of Domain Adaptive Imitation Learning (DAIL)☆24Oct 26, 2020Updated 5 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Mar 30, 2021Updated 4 years ago
- ☆21Dec 8, 2022Updated 3 years ago
- Port-Hamiltonian Approach to Neural Network Training☆27Dec 11, 2019Updated 6 years ago
- ICML 2020 Paper: Latent Variable Modelling with Hyperbolic Normalizing Flows☆54Dec 8, 2022Updated 3 years ago
- Official code for ICML 2020 paper "Variational Bayesian Quantization"☆24Dec 8, 2022Updated 3 years ago
- Code for T-MARS data filtering☆35Aug 23, 2023Updated 2 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Feb 3, 2020Updated 6 years ago
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Jun 30, 2020Updated 5 years ago
- Code for "Accelerating Natural Gradient with Higher-Order Invariance"☆30Jun 28, 2019Updated 6 years ago
- Automatic Functional Differentiation in JAX☆81Sep 18, 2025Updated 5 months ago
- Tools for visualizing and comparing data from vertebrate retinas☆14Jan 20, 2025Updated last year
- Code repository for "Multi-Task Encoder-Dual-Decoder Modeling Framework on Mixed Frequency Data", International Journal of Forecasting, 2…☆12Feb 18, 2024Updated 2 years ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆37Mar 18, 2025Updated 11 months ago
- Gradient-based constrained optimization for JAX☆37Sep 14, 2022Updated 3 years ago
- Normalizing flows in PyTorch☆444Dec 1, 2025Updated 3 months ago
- AlgoPy is a Research Prototype for Algorithmic Differentation in Python☆88Jul 7, 2024Updated last year