ryushinn / flows-on-sphereLinks
This is a Pytorch implementation of [normalizing flows on tori and spheres, ICML 2020]
☆19Updated 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:
- This is project page for the paper "RG-Flow: a hierarchical and explainable flow model based on renormalization group and sparse prior". …☆89Updated 3 years ago
- Forward mode laplacian implemented in JAX tracer☆29Updated last month
- Continuous-time gradient flow for generative modeling and variational inference☆33Updated 7 years ago
- Pytorch source code for arXiv paper Neural Network Renormalization Group, a generative model using variational renormalization group and …☆79Updated 6 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆34Updated last year
- Things that make me feel productive☆15Updated 3 years ago
- Reverse-mode AD of dominant sparse eigensolver using Pytorch.☆41Updated 5 years ago
- JAX exponential map normalising flows on sphere☆17Updated 5 years ago
- ☆33Updated 5 years ago
- Machine learning Calabi-Yau metrics with JAX☆17Updated 2 years ago
- Normalizing Flows using JAX☆85Updated last year
- A library for programmatically generating equivariant layers through constraint solving☆277Updated 2 years ago
- Differentiable scientific computing library☆156Updated last year
- ☆16Updated 3 years ago
- Matrix-free linear algebra in JAX.☆148Updated this week
- Pytorch implement of the paper Neural Canonical Transformation with Symplectic Flows☆30Updated 5 years ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Updated 3 years ago
- Turning SymPy expressions into PyTorch modules.☆151Updated 2 years ago
- Orthogonal polynomials with JAX☆26Updated last week
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …☆51Updated 3 years ago
- Hierarchical variational models for physics.☆18Updated 5 years ago
- Rescure useful code in TensorNetwork by google for further personal usage.☆12Updated 11 months ago
- Differentiable interface to Firedrake for JAX☆15Updated 4 years ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆35Updated 2 years ago
- Implementation of Forward Laplacian algorithm in JAX☆88Updated last month
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Updated 3 years ago
- A differentiation API for PyTorch☆30Updated 5 years ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago