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.
  ☆632Updated last week
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆133Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆283Updated this week
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆372Updated 4 months ago
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX and Flax.
  ☆529Updated this week
• wilson-labs / cola
  Compositional Linear Algebra
  ☆490Updated last month
• danielward27 / flowjax
  ☆182Updated 3 weeks ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆464Updated 2 weeks ago
• probabilists / zuko
  Normalizing flows in PyTorch
  ☆402Updated 2 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆94Updated 3 months ago
• probabilisticai / probai-2022
  Materials of the Nordic Probabilistic AI School 2022.
  ☆180Updated 2 years ago
• QUVA-Lab / escnn
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆463Updated 10 months ago
• google-deepmind / distrax
  ☆589Updated last week
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆273Updated 2 years ago
• probml / dynamax
  A Python package for probabilistic state space modeling with JAX
  ☆861Updated last month
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆20Updated last year
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆104Updated last year
• google / grain
  Library for reading and processing ML training data.
  ☆519Updated this week
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆282Updated last month
• blackjax-devs / blackjax
  BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
  ☆947Updated 4 months ago
• J-zin / Awesome-Diffusion-Flow-Samplers
  Collecting research materials on neural samplers with diffusion/flow models
  ☆58Updated last month
• AndreasBergmeister / pjax
  A projection-based framework for gradient-free and parallel learning
  ☆104Updated 2 months ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆400Updated 4 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆292Updated last year
• malbergo / stochastic-interpolants
  ☆172Updated last year
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆113Updated 8 months ago
• nmboffi / jax-interpolants
  Definitive implementation of the stochastic interpolant framework for generative modeling in jax.
  ☆30Updated last month
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆106Updated last month
• MachineLearningLifeScience / stochman
  Algorithms for computations on random manifolds made easier
  ☆92Updated last year
• mila-iqia / ResearchTemplate
  Research Project Template Repository
  ☆33Updated last week