Alternatives and similar repositories for zuko

Users that are interested in zuko are comparing it to the libraries listed below

• VincentStimper / normalizing-flows
  PyTorch implementation of normalizing flow models
  ☆895Updated last year
• bayesiains / nflows
  Normalizing flows in PyTorch
  ☆963Updated 9 months ago
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆309Updated 4 months ago
• johannbrehmer / manifold-flow
  Manifold-learning flows (ℳ-flows)
  ☆230Updated 4 years ago
• martenlienen / torchode
  A parallel ODE solver for PyTorch
  ☆270Updated last year
• LukasRinder / normalizing-flows
  Implementation of normalizing flows in TensorFlow 2 including a small tutorial.
  ☆147Updated 3 weeks ago
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
  ☆662Updated this week
• AdamCobb / hamiltorch
  PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks
  ☆459Updated last year
• patrick-kidger / torchcde
  Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
  ☆453Updated last month
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆521Updated 6 months ago
• malbergo / stochastic-interpolants
  ☆180Updated last year
• alisiahkoohi / Langevin-dynamics
  Sampling with gradient-based Markov Chain Monte Carlo approaches
  ☆108Updated last year
• bayesiains / nsf
  Code for Neural Spline Flows paper
  ☆279Updated 5 years ago
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX and Flax.
  ☆543Updated this week
• probabilists / lampe
  Likelihood-free AMortized Posterior Estimation with PyTorch
  ☆131Updated last year
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆115Updated 3 weeks ago
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆72Updated 11 months ago
• patrick-kidger / torchcubicspline
  Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…
  ☆270Updated 3 years ago
• tonyduan / normalizing-flows
  Neural Spline Flow, RealNVP, Autoregressive Flow, 1x1Conv in PyTorch.
  ☆280Updated last year
• danielward27 / flowjax
  ☆192Updated this week
• didriknielsen / survae_flows
  Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"
  ☆288Updated 4 years ago
• janosh / awesome-normalizing-flows
  Awesome resources on normalizing flows.
  ☆1,571Updated 3 months ago
• xwinxu / bayeSDE
  Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
  ☆170Updated 3 years ago
• jacobjinkelly / easy-neural-ode
  Code for the paper "Learning Differential Equations that are Easy to Solve"
  ☆284Updated 3 years ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆145Updated 4 months ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆103Updated 9 months ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆481Updated this week
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆78Updated 3 years ago
• rtqichen / ffjord
  code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".
  ☆657Updated 5 years ago
• angusphillips / particle_denoising_diffusion_sampler
  Code repository for Particle Denoising Diffusion Sampler
  ☆15Updated last year