klieret / wandb-offline-sync-hook
A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!
☆72Updated 2 weeks ago
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☆67Updated last month
- Use Jax functions in Pytorch☆238Updated last year
- NF-Layers for constructing neural functionals.☆82Updated last year
- Code for the paper: Rotating Features for Object Discovery☆50Updated 7 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆67Updated 3 weeks ago
- Flow-matching algorithms in JAX☆86Updated 7 months ago
- Modern Fixed Point Systems using Pytorch☆88Updated last year
- code for "Riemannian Flow Matching on General Geometries".☆206Updated last year
- ☆23Updated 3 months ago
- Annotated Flow Matching paper☆175Updated 6 months ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆138Updated 5 months ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆45Updated 2 years ago
- ☆24Updated 7 months ago
- Educational implementation of the Discrete Flow Matching paper☆79Updated 6 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆55Updated last year
- Example of how to use Weights & Biases on Slurm☆113Updated 2 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated last week
- ☆58Updated 9 months ago
- ☆138Updated last year
- Pytorch-like dataloaders for JAX.☆76Updated 5 months ago
- This is the official source code for SLATE. We provide the code for the model, the training code, and a dataset loader for the 3D Shapes …☆84Updated 2 years ago
- Run PyTorch in JAX. 🤝☆231Updated last month
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆338Updated 2 years ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models☆67Updated last year
- Score-based generative models for compact manifolds☆108Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- ☆126Updated last year
- Efficient Video Prediction via Sparsely Conditioned Flow Matching. In ICCV, 2023.☆13Updated 9 months ago
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆50Updated last month