klieret / wandb-offline-sync-hook
A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!
☆74Updated last week
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
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆73Updated 2 months ago
- Use Jax functions in Pytorch☆239Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆74Updated last month
- Flow-matching algorithms in JAX☆89Updated 8 months ago
- code for "Riemannian Flow Matching on General Geometries".☆218Updated last year
- Modern Fixed Point Systems using Pytorch☆89Updated last year
- NF-Layers for constructing neural functionals.☆82Updated last year
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆46Updated 2 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- Educational implementation of the Discrete Flow Matching paper☆83Updated 7 months ago
- ☆28Updated 8 months ago
- Annotated Flow Matching paper☆181Updated 7 months ago
- WandB sweeps integration with Hydra sweeper☆47Updated last year
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆140Updated 6 months ago
- ☆156Updated 8 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆57Updated last year
- Pytorch-like dataloaders for JAX.☆80Updated 6 months ago
- Code for the paper: Rotating Features for Object Discovery☆50Updated 8 months ago
- ☆140Updated last year
- A simple library for scaling up JAX programs☆134Updated 5 months ago
- A curated list for awesome discrete diffusion models resources.☆296Updated 2 months ago
- ☆59Updated 10 months ago
- Example of how to use Weights & Biases on Slurm☆114Updated 2 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated last month
- Run PyTorch in JAX. 🤝☆236Updated 2 months ago
- ☆23Updated 2 weeks ago
- Simple Guidance Mechanisms for Discrete Diffusion Models☆31Updated 4 months ago
- ☆127Updated last year
- Code for the paper https://arxiv.org/abs/2205.14987v2☆49Updated last year
- Collecting research materials on neural samplers with diffusion/flow models☆45Updated last month