elyall / wandb_on_slurmLinks
Example of how to use Weights & Biases on Slurm
☆115Updated 2 years ago
Alternatives and similar repositories for wandb_on_slurm
Users that are interested in wandb_on_slurm are comparing it to the libraries listed below
Sorting:
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- WandB sweeps integration with Hydra sweeper☆49Updated last year
- Research Project Template Repository☆32Updated 2 weeks ago
- Tools to connect to and interact with the Mila cluster☆72Updated last week
- SEML: Slurm Experiment Management Library☆182Updated last month
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆31Updated last month
- Implementation of a list sweeper: instead of the cartesian product, sweep over the zipped list☆23Updated 3 months ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆208Updated last year
- Use Jax functions in Pytorch☆242Updated last year
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆81Updated 3 weeks ago
- Optimal Transport Dataset Distance☆166Updated 3 years ago
- JAX Arrays for human consumption☆93Updated this week
- CLU lets you write beautiful training loops in JAX.☆346Updated this week
- ☆157Updated last year
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆198Updated 10 months ago
- Efficiently Composable Data Augmentation on the GPU with Jax☆37Updated last month
- ☆208Updated 2 years ago
- Running Jax in PyTorch Lightning☆102Updated 6 months ago
- TUI for the Slurm Workload Manager☆179Updated last month
- Pytorch-like dataloaders for JAX.☆87Updated 3 weeks ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆55Updated last year
- Aggregate multiple tensorboard runs to new summary or csv files☆173Updated 3 weeks ago
- ☆339Updated 3 months ago
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
- ☆99Updated 3 years ago
- A simple command line tool to show GPU usage on a SLURM cluster☆109Updated last year
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- NF-Layers for constructing neural functionals.☆85Updated last year
- A library for programmatically generating equivariant layers through constraint solving☆268Updated 2 years ago