ryushinn / flows-on-sphereLinks
This is a Pytorch implementation of [normalizing flows on tori and spheres, ICML 2020]
☆18Updated 3 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:
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆34Updated last year
- This is project page for the paper "RG-Flow: a hierarchical and explainable flow model based on renormalization group and sparse prior". …☆88Updated 3 years ago
- Continuous-time gradient flow for generative modeling and variational inference☆33Updated 7 years ago
- Matrix-free linear algebra in JAX.☆141Updated 3 weeks ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆35Updated 2 years ago
- A library for programmatically generating equivariant layers through constraint solving☆276Updated 2 years ago
- Pytorch source code for arXiv paper Neural Network Renormalization Group, a generative model using variational renormalization group and …☆79Updated 6 years ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 3 weeks ago
- S2FFT: Differentiable and accelerated spherical transforms☆207Updated this week
- Differentiable scientific computing library☆155Updated last year
- Exact OU processes with JAX☆54Updated 6 months ago
- JAX exponential map normalising flows on sphere☆17Updated 5 years ago
- Things that make me feel productive☆15Updated 3 years ago
- Automatic Functional Differentiation in JAX☆78Updated 3 weeks ago
- Normalizing Flows using JAX☆84Updated last year
- Turning SymPy expressions into PyTorch modules.☆151Updated 2 years ago
- ☆16Updated 3 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆101Updated this week
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 7 months ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Simulation-based inference benchmark☆103Updated 8 months ago
- Differentiable interface to Firedrake for JAX☆15Updated 4 years ago
- Riemannian Optimization Using JAX☆53Updated last year
- Experiments for the AISTATS publication on Reparameterizing Distributions over Lie Groups☆52Updated 3 years ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆171Updated 3 years ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆131Updated last year
- ☆22Updated 5 years ago
- A differentiation API for PyTorch☆30Updated 5 years ago
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆29Updated 4 months ago