Alternatives and similar repositories for wandb-offline-sync-hook

Users that are interested in wandb-offline-sync-hook are comparing it to the libraries listed below

• willisma / diffuse_nnx
  A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…
  ☆125Updated 2 months ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆259Updated 2 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆112Updated last year
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆149Updated 10 months ago
• locuslab / torchdeq
  Modern Fixed Point Systems using Pytorch
  ☆125Updated 2 years ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆106Updated 2 months ago
• AllanYangZhou / nfn
  NF-Layers for constructing neural functionals.
  ☆93Updated last year
• kvfrans / splus
  ☆122Updated 6 months ago
• facebookresearch / riemannian-fm
  code for "Riemannian Flow Matching on General Geometries".
  ☆278Updated last year
• nmboffi / flow-maps
  Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).
  ☆61Updated 2 months ago
• liusulin / DDPD
  Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]
  ☆84Updated 8 months ago
• microsoft / soc-fine-tuning-sd
  Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching
  ☆58Updated 6 months ago
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆224Updated last year
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆202Updated 6 months ago
• DoronHav / WassersteinFlowMatching
  ☆36Updated this week
• raywang4 / EqM
  ☆170Updated 2 months ago
• malbergo / stochastic-interpolants
  ☆198Updated last year
• yilundu / reduce_reuse_recycle
  ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC
  ☆147Updated last year
• brianlck / FlexMDM
  ☆34Updated 3 months ago
• wpeebles / G.pt
  Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"
  ☆345Updated 3 years ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆97Updated last week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆310Updated 2 months ago
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆130Updated 5 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆86Updated 11 months ago
• loeweX / RotatingFeatures
  Code for the paper: Rotating Features for Object Discovery
  ☆53Updated last year
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆126Updated last year
• kuleshov-group / discrete-diffusion-guidance
  Simple Guidance Mechanisms for Discrete Diffusion Models
  ☆64Updated last year
• lucmos / relreps
  Relative representations can be leveraged to enable solving tasks regarding "latent communication": from zero-shot model stitching to lat…
  ☆64Updated 2 years ago
• google-deepmind / functa
  ☆158Updated last year
• nmboffi / jax-interpolants
  Definitive implementation of the stochastic interpolant framework for generative modeling in jax.
  ☆39Updated 5 months ago