klieret / wandb-offline-sync-hook
A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!
☆49Updated last week
Related projects: ⓘ
- Use Jax functions in Pytorch☆224Updated last year
- code for "Riemannian Flow Matching on General Geometries".☆166Updated 6 months ago
- Flow-matching algorithms in JAX☆62Updated last month
- Code for the paper: Rotating Features for Object Discovery☆46Updated last month
- Annotated Flow Matching paper☆91Updated this week
- Score-based generative models for compact manifolds☆99Updated 8 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆46Updated 6 months ago
- ☆199Updated last year
- Code for the paper https://arxiv.org/abs/2205.14987v2☆40Updated 5 months ago
- NF-Layers for constructing neural functionals.☆72Updated 8 months ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆109Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆154Updated 11 months ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆30Updated 5 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆15Updated 3 weeks ago
- Example of how to use Weights & Biases on Slurm☆108Updated 2 years ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models☆64Updated 10 months ago
- Pytorch-like dataloaders in JAX.☆52Updated last month
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆138Updated 2 years ago
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.☆14Updated last month
- Modern Fixed Point Systems using Pytorch☆77Updated 10 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆48Updated 3 weeks ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆128Updated 11 months ago
- ☆131Updated last year
- A summary of related works about flow matching, stochastic interpolants☆274Updated last month
- A simple implimentation of Bayesian Flow Networks (BFN)☆236Updated 8 months ago
- Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(Lightning)☆89Updated 3 years ago
- Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022☆71Updated 2 years ago
- ☆290Updated 2 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆68Updated 3 months ago
- PyTorch implementation of Variational Diffusion Models.☆69Updated 4 months ago