ryushinn / flows-on-sphereLinks
This is a Pytorch implementation of [normalizing flows on tori and spheres, ICML 2020]
☆18Updated 2 years ago
Alternatives and similar repositories for flows-on-sphere
Users that are interested in flows-on-sphere are comparing it to the libraries listed below
Sorting:
- This is project page for the paper "RG-Flow: a hierarchical and explainable flow model based on renormalization group and sparse prior". …☆87Updated 3 years ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 6 months ago
- Pytorch source code for arXiv paper Neural Network Renormalization Group, a generative model using variational renormalization group and …☆78Updated 5 years ago
- Normalizing Flows using JAX☆84Updated last year
- Machine learning Calabi-Yau metrics with JAX☆17Updated 2 years ago
- Code for Gaussian Score Matching Variational Inference☆34Updated 6 months ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆35Updated 2 years ago
- Matrix-free linear algebra in JAX.☆138Updated last week
- A library for programmatically generating equivariant layers through constraint solving☆273Updated 2 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆34Updated last year
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Simulation-based inference benchmark☆101Updated 7 months ago
- Continuous-time gradient flow for generative modeling and variational inference☆33Updated 6 years ago
- A generic interface for linear algebra backends☆73Updated 6 months ago
- Things that make me feel productive☆15Updated 2 years ago
- Riemannian Optimization Using JAX☆52Updated last year
- Rescure useful code in TensorNetwork by google for further personal usage.☆12Updated 8 months ago
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆51Updated 3 years ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆171Updated 3 years ago
- Exact OU processes with JAX☆53Updated 5 months ago
- ☆16Updated 3 years ago
- Implementation of Forward Laplacian algorithm in JAX☆81Updated last week
- Hierarchical variational models for physics.☆17Updated 5 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Differentiable interface to Firedrake for JAX☆15Updated 4 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated last year
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆100Updated 3 weeks ago
- ☆274Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago