probabilists / zuko

Related projects ⓘ

Alternatives and complementary repositories for zuko

• VincentStimper / normalizing-flows
  PyTorch implementation of normalizing flow models
  ☆723Updated 2 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…
  ☆301Updated 3 weeks ago
• bayesiains / nflows
  Normalizing flows in PyTorch
  ☆848Updated last year
• johannbrehmer / manifold-flow
  Manifold-learning flows (ℳ-flows)
  ☆230Updated 4 years ago
• martenlienen / torchode
  A parallel ODE solver for PyTorch
  ☆232Updated last month
• bayesiains / nsf
  Code for Neural Spline Flows paper
  ☆259Updated 4 years ago
• LukasRinder / normalizing-flows
  Implementation of normalizing flows in TensorFlow 2 including a small tutorial.
  ☆144Updated 2 months ago
• tonyduan / normalizing-flows
  Neural Spline Flow, RealNVP, Autoregressive Flow, 1x1Conv in PyTorch.
  ☆272Updated 11 months ago
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆526Updated this week
• didriknielsen / survae_flows
  Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"
  ☆284Updated 3 years ago
• probabilists / lampe
  Likelihood-free AMortized Posterior Estimation with PyTorch
  ☆119Updated 2 months ago
• AdamCobb / hamiltorch
  PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks
  ☆425Updated 2 months ago
• patrick-kidger / torchcde
  Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
  ☆421Updated last year
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX.
  ☆461Updated 2 weeks ago
• gpapamak / maf
  Masked Autoregressive Flow
  ☆204Updated 3 months ago
• xwinxu / bayeSDE
  Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
  ☆170Updated 2 years ago
• danielward27 / flowjax
  ☆107Updated 3 weeks ago
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆94Updated 2 months ago
• JTT94 / diffusion_schrodinger_bridge
  PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.
  ☆146Updated 2 years ago
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆471Updated this week
• acids-ircam / pytorch_flows
  Implementation and tutorials of normalizing flows with the novel distributions module
  ☆160Updated 4 years ago
• addtt / variational-diffusion-models
  PyTorch implementation of Variational Diffusion Models.
  ☆77Updated 6 months ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆332Updated this week
• janosh / awesome-normalizing-flows
  Awesome resources on normalizing flows.
  ☆1,438Updated last month
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆107Updated this week
• alisiahkoohi / Langevin-dynamics
  Sampling with gradient-based Markov Chain Monte Carlo approaches
  ☆89Updated 7 months ago
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆661Updated 2 years ago
• kamenbliznashki / normalizing_flows
  Pytorch implementations of density estimation algorithms: BNAF, Glow, MAF, RealNVP, planar flows
  ☆606Updated 3 years ago
• CW-Huang / sdeflow-light
  A minimalist implementation of score-based diffusion model
  ☆121Updated 3 years ago
• rtqichen / ffjord
  code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".
  ☆629Updated 4 years ago