captain-pool / hydra-wandb-sweeperLinks
WandB sweeps integration with Hydra sweeper
☆50Updated last year
Alternatives and similar repositories for hydra-wandb-sweeper
Users that are interested in hydra-wandb-sweeper are comparing it to the libraries listed below
Sorting:
- Running Jax in PyTorch Lightning☆118Updated last year
- Example of how to use Weights & Biases on Slurm☆118Updated 3 years ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated last year
- A functional training loops library for JAX☆88Updated last year
- ☆118Updated last month
- A Python package of computer vision models for the Equinox ecosystem.☆110Updated last year
- Building blocks for productive research☆67Updated last week
- ☆56Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- ☆162Updated 2 years ago
- Efficiently Composable Data Augmentation on the GPU with Jax☆42Updated 8 months ago
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)☆52Updated last week
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Lightweight ML Experiment Logging 📖☆80Updated last year
- Pytorch-like dataloaders for JAX.☆98Updated last month
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆35Updated last week
- JAX Arrays for human consumption☆110Updated 2 months ago
- CLU lets you write beautiful training loops in JAX.☆366Updated last week
- Unofficial JAX implementations of deep learning research papers☆160Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- A collection of meta-learning algorithms in Jax☆24Updated 3 years ago
- Run PyTorch in JAX. 🤝☆311Updated 3 months ago
- Open source code for EigenGame.☆34Updated 2 years ago
- LoRA for arbitrary JAX models and functions☆143Updated last year
- Easy Hypernetworks in Pytorch and Jax☆106Updated 2 years ago
- Use Jax functions in Pytorch☆259Updated 2 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 3 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆118Updated 3 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆208Updated last year