wandb / sweeps
W&B Hyperparameter Sweep Engine. File sweeps related issues at the W&B client: https://github.com/wandb/client
☆34Updated 2 months ago
Related projects: ⓘ
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated 11 months ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Updated last year
- Convert scikit-learn models to PyTorch modules☆157Updated 4 months ago
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆74Updated this week
- Lightweight Hyperparameter Optimization 🚂☆145Updated 3 weeks ago
- JMP is a Mixed Precision library for JAX.☆183Updated 4 months ago
- Unofficial JAX implementations of deep learning research papers☆150Updated 2 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- ☆64Updated 10 months ago
- Multidimensional indexing for tensors☆108Updated last year
- Torch Distributed Experimental☆115Updated last month
- My implementation of DeepMind's Perceiver☆63Updated 3 years ago
- ☆40Updated this week
- Running Jax in PyTorch Lightning☆76Updated last year
- a lightweight transformer library for PyTorch☆71Updated 2 years ago
- JAX Synergistic Memory Inspector☆161Updated 2 months ago
- A Pytree Module system for Deep Learning in JAX☆215Updated last year
- Distributed skorch on Ray Train☆57Updated 2 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated 10 months ago
- Functional deep learning☆106Updated last year
- Lightweight ML Experiment Logging 📖☆79Updated 3 weeks ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆184Updated 2 years ago
- Lightning HPO & Training Studio App☆17Updated last year
- A collection of code snippets for my PyTorch Lightning projects☆108Updated 3 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆101Updated last year
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Differentiable Sorting Networks☆103Updated last year
- Check if you have training samples in your test set☆64Updated 2 years ago
- ☆182Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆59Updated last year