Alternatives and similar repositories for flows-on-sphere

Users that are interested in flows-on-sphere are comparing it to the libraries listed below

• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆84Updated last year
• probabilistic-numerics / probnum-spring-school
  Tutorial materials of the Probabilistic Numerics Spring School.
  ☆35Updated 2 years ago
• johannbrehmer / pointers
  Things that make me feel productive
  ☆15Updated 2 years ago
• li012589 / NeuralRG
  Pytorch source code for arXiv paper Neural Network Renormalization Group, a generative model using variational renormalization group and …
  ☆78Updated 5 years ago
• vdutor / SphericalHarmonics
  Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax
  ☆33Updated last year
• modichirag / GSM-VI
  Code for Gaussian Score Matching Variational Inference
  ☆34Updated 5 months ago
• lezcano / expm
  Differentiable and numerically stable implementation of the matrix exponential
  ☆33Updated 4 years ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆133Updated last week
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆271Updated 2 years ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆100Updated 6 months ago
• thoglu / jammy_flows
  A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…
  ☆43Updated 4 months ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated 2 years ago
• hongyehu / RG-Flow
  This is project page for the paper "RG-Flow: a hierarchical and explainable flow model based on renormalization group and sparse prior". …
  ☆87Updated 2 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆104Updated last year
• dimitra-maoutsa / odes_for_sdes
  Deterministic particle dynamics for simulating Fokker-Planck probability flows
  ☆26Updated 2 years ago
• ml4physics / cyjax
  Machine learning Calabi-Yau metrics with JAX
  ☆17Updated 2 years ago
• y1xiaoc / fwdlap
  Forward mode laplacian implemented in JAX tracer
  ☆29Updated 5 months ago
• IvanYashchuk / jax-firedrake
  Differentiable interface to Firedrake for JAX
  ☆14Updated 4 years ago
• necludov / iMCMC
  ☆22Updated 5 years ago
• xwinxu / bayeSDE
  Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
  ☆171Updated 3 years ago
• wangleiphy / MongeAmpereFlow
  Continuous-time gradient flow for generative modeling and variational inference
  ☆32Updated 6 years ago
• astro-informatics / s2fft
  S2FFT: Differentiable and accelerated spherical transforms
  ☆199Updated this week
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• patrick-kidger / sympytorch
  Turning SymPy expressions into PyTorch modules.
  ☆148Updated 2 years ago
• kazewong / flowMC
  Normalizing-flow enhanced sampling package for probabilistic inference in Jax
  ☆250Updated last week
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆74Updated 4 months ago
• dingo-gw / flow-matching-posterior-estimation
  ☆27Updated last year
• gregversteeg / esh_dynamics
  Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling
  ☆36Updated 3 years ago
• iaifi / summer-school-2022
  ☆16Updated 3 years ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆100Updated this week