Alternatives and similar repositories for diffusion_journal_club_24

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

• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
  ☆685Updated last month
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆309Updated 2 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆141Updated 2 years ago
• danielward27 / flowjax
  ☆205Updated 2 months ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆524Updated last week
• thomaspinder / GPJax
  Gaussian processes in JAX and Flax.
  ☆569Updated 3 weeks ago
• google-deepmind / distrax
  ☆613Updated last month
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆379Updated this week
• probabilists / zuko
  Normalizing flows in PyTorch
  ☆438Updated last month
• wilson-labs / cola
  Compositional Linear Algebra
  ☆504Updated 5 months ago
• homerjed / sbgm
  Score-based Diffusion models in JAX.
  ☆15Updated last week
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆80Updated last year
• jax-ml / jax-ai-stack
  ☆261Updated 2 weeks ago
• MachineLearningLifeScience / stochman
  Algorithms for computations on random manifolds made easier
  ☆94Updated 2 years ago
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆20Updated last year
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆303Updated 3 weeks ago
• probabilisticai / probai-2022
  Materials of the Nordic Probabilistic AI School 2022.
  ☆181Updated 3 years ago
• malbergo / stochastic-interpolants
  ☆200Updated last year
• jeremiecoullon / jax-tqdm
  Add a tqdm progress bar to your JAX scans and loops.
  ☆124Updated 8 months ago
• baumgach / tue-slurm-helloworld
  Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container
  ☆33Updated 2 years ago
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆363Updated 2 weeks ago
• J-zin / Awesome-Diffusion-Flow-Samplers
  Collecting research materials on neural samplers with diffusion/flow models
  ☆59Updated 6 months ago
• QUVA-Lab / escnn
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆493Updated last year
• alonfnt / pcax
  Differentiable Principal Component Analysis (PCA) implementation in JAX
  ☆30Updated 8 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆97Updated 3 weeks ago
• smsharma / awesome-neural-sbi
  Community-sourced list of papers and resources on neural simulation-based inference.
  ☆139Updated this week
• probml / dynamax
  A Python package for probabilistic state space modeling with JAX
  ☆903Updated this week
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆531Updated 8 months ago
• andres-fr / skerch
  Sketched linear operations for PyTorch
  ☆100Updated 2 months ago
• AndreasBergmeister / pjax
  A projection-based framework for gradient-free and parallel learning
  ☆109Updated 6 months ago