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.
  ☆676Updated 3 weeks ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆503Updated 4 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆309Updated 2 months ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆519Updated this week
• thomaspinder / GPJax
  Gaussian processes in JAX and Flax.
  ☆563Updated this week
• google-deepmind / distrax
  ☆612Updated last week
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆140Updated 2 years ago
• probabilists / zuko
  Normalizing flows in PyTorch
  ☆433Updated 2 weeks ago
• danielward27 / flowjax
  ☆205Updated last month
• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆377Updated 2 months ago
• rbalestr-lab / stable-pretraining
  Reliable, minimal and scalable library for pretraining foundation and world models
  ☆110Updated last month
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆97Updated this week
• jax-ml / jax-ai-stack
  ☆255Updated this week
• AndreasBergmeister / pjax
  A projection-based framework for gradient-free and parallel learning
  ☆108Updated 6 months ago
• homerjed / sbgm
  Score-based Diffusion models in JAX.
  ☆15Updated this week
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆75Updated last year
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆20Updated last year
• baumgach / tue-slurm-helloworld
  Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container
  ☆33Updated 2 years ago
• CLAIRE-Labo / python-ml-research-template
  A template for starting reproducible Python machine-learning projects with hardware acceleration. Find an example at https://github.com/C…
  ☆113Updated 6 months ago
• probml / dynamax
  A Python package for probabilistic state space modeling with JAX
  ☆896Updated 2 weeks ago
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆298Updated this week
• acerbilab / amortized-conditioning-engine
  Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
  ☆21Updated 5 months ago
• MachineLearningLifeScience / stochman
  Algorithms for computations on random manifolds made easier
  ☆94Updated 2 years ago
• malbergo / stochastic-interpolants
  ☆198Updated last year
• probabilisticai / probai-2022
  Materials of the Nordic Probabilistic AI School 2022.
  ☆181Updated 3 years ago
• smsharma / awesome-neural-sbi
  Community-sourced list of papers and resources on neural simulation-based inference.
  ☆138Updated last month
• probabilisticai / probai-2023
  Materials of the Nordic Probabilistic AI School 2023.
  ☆91Updated 2 years ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆110Updated last month
• google / grain
  Library for reading and processing ML training data.
  ☆626Updated this week
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆147Updated last month