probabilists / zuko

Related projects ⓘ

Alternatives and complementary repositories for zuko

• VincentStimper / normalizing-flows
  PyTorch implementation of normalizing flow models
  ☆713Updated 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 last week
• bayesiains / nflows
  Normalizing flows in PyTorch
  ☆845Updated 11 months ago
• johannbrehmer / manifold-flow
  Manifold-learning flows (ℳ-flows)
  ☆230Updated 3 years ago
• martenlienen / torchode
  A parallel ODE solver for PyTorch
  ☆225Updated last month
• LukasRinder / normalizing-flows
  Implementation of normalizing flows in TensorFlow 2 including a small tutorial.
  ☆141Updated 2 months ago
• probabilists / lampe
  Likelihood-free AMortized Posterior Estimation with PyTorch
  ☆119Updated 2 months ago
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆521Updated this week
• tonyduan / normalizing-flows
  Neural Spline Flow, RealNVP, Autoregressive Flow, 1x1Conv in PyTorch.
  ☆270Updated 10 months ago
• patrick-kidger / torchcde
  Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
  ☆420Updated last year
• bayesiains / nsf
  Code for Neural Spline Flows paper
  ☆259Updated 4 years ago
• didriknielsen / survae_flows
  Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"
  ☆283Updated 3 years ago
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX.
  ☆459Updated last week
• AdamCobb / hamiltorch
  PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks
  ☆422Updated 2 months ago
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆468Updated last month
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆93Updated 2 months ago
• malbergo / stochastic-interpolants
  ☆104Updated 8 months ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆107Updated this week
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆330Updated last week
• TyXe-BDL / TyXe
  ☆145Updated 2 years ago
• gpapamak / maf
  Masked Autoregressive Flow
  ☆203Updated 3 months ago
• danielward27 / flowjax
  ☆101Updated 2 weeks ago
• alisiahkoohi / Langevin-dynamics
  Sampling with gradient-based Markov Chain Monte Carlo approaches
  ☆88Updated 6 months ago
• xwinxu / bayeSDE
  Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
  ☆170Updated 2 years ago
• JTT94 / diffusion_schrodinger_bridge
  PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.
  ☆144Updated 2 years ago
• YannDubs / Neural-Process-Family
  Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.
  ☆202Updated 4 months ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆89Updated 7 months ago
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆52Updated this week
• google-research / vdm
  ☆298Updated 2 years ago