elyall / wandb_on_slurm
Example of how to use Weights & Biases on Slurm
☆108Updated 2 years ago
Related projects: ⓘ
- A Machine Learning workflow for Slurm.☆145Updated 3 years ago
- SEML: Slurm Experiment Management Library☆166Updated last week
- Tools to connect to and interact with the Mila cluster☆60Updated last month
- WandB sweeps integration with Hydra sweeper☆45Updated 7 months ago
- 🧶 Modular VAE disentanglement framework for python built with PyTorch Lightning ▸ Including metrics and datasets ▸ With strongly supervi…☆122Updated last year
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆199Updated last year
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆49Updated last week
- ☆258Updated 6 months ago
- Use Jax functions in Pytorch☆224Updated last year
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆173Updated last month
- A library for uncertainty quantification based on PyTorch☆118Updated 2 years ago
- ☆19Updated 4 years ago
- ☆182Updated last year
- ☆149Updated 9 months ago
- GFlowNet library☆213Updated 3 weeks ago
- ☆50Updated last year
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆50Updated last year
- ☆104Updated last week
- Codebase for Learning Invariances in Neural Networks☆94Updated last year
- TUI for the Slurm Workload Manager☆118Updated 2 weeks ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆555Updated 4 months ago
- A simple command line tool to show GPU usage on a SLURM cluster☆101Updated 5 months ago
- A minimalist implementation of score-based diffusion model☆120Updated 3 years ago
- ☆96Updated 2 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆300Updated 6 months ago
- Optimal Transport Dataset Distance☆151Updated 2 years ago
- Bayesianize: A Bayesian neural network wrapper in pytorch☆81Updated 4 months ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Agustinus' very opiniated publication-ready plotting library☆51Updated 2 weeks ago
- A curated list of techniques to avoid posterior collapse☆85Updated last year