bamos / setGPU
Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.
☆107Updated 2 years ago
Alternatives and similar repositories for setGPU:
Users that are interested in setGPU are comparing it to the libraries listed below
- Python bindings for pyNVML and psutil library over network☆50Updated last year
- Code for paper "L4: Practical loss-based stepsize adaptation for deep learning"☆124Updated 5 years ago
- Implementation and demonstration of backdrop in pytorch. Code and demonstration of GP dataset generator.☆68Updated 6 years ago
- Forward-mode Automatic Differentiation for TensorFlow☆140Updated 6 years ago
- Differentiable bitonic sorting☆138Updated 4 years ago
- Weight initialization schemes for PyTorch nn.Modules☆70Updated 7 years ago
- a lightweight and simple logger for Machine Learning☆127Updated 4 years ago
- Decoupled Neural Interfaces using Synthetic Gradients for PyTorch☆236Updated 6 years ago
- Second-order optimiser for deep networks☆76Updated 6 years ago
- Probabilistic Programming and Statistical Inference in PyTorch☆110Updated 7 years ago
- ☆220Updated 6 years ago
- ☆132Updated 7 years ago
- Python library for extracting mini-batches of data from a data source for the purpose of training neural networks☆87Updated 5 years ago
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆183Updated last year
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆148Updated 7 years ago
- Evaluation code with models for the paper "On the Quantitative Analysis of Decoder-Based Generative Models"☆130Updated 7 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- A Python implementation of the gradient REBAR estimator.☆46Updated 6 years ago
- Tools for PyTorch☆221Updated 2 years ago
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆65Updated 7 years ago
- Research code need not be ugly.☆77Updated last year
- ☆36Updated 5 years ago
- ICLR Reproducibility Challenge 2019☆219Updated 5 years ago
- Odds and Ends and Things I've implemented.☆78Updated 5 years ago
- Plot TensorBoard graphs fast☆51Updated 3 years ago
- Distributed Learning by Pair-Wise Averaging☆53Updated 7 years ago
- An extension to Sacred for automated hyperparameter optimization.☆59Updated 6 years ago
- A utility library around PyTorch☆244Updated 2 years ago
- ☆79Updated 6 years ago