bamos / setGPULinks
Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.
☆111Updated 3 years ago
Alternatives and similar repositories for setGPU
Users that are interested in setGPU are comparing it to the libraries listed below
Sorting:
- Forward-mode Automatic Differentiation for TensorFlow☆139Updated 7 years ago
- ☆219Updated 7 years ago
- Code for paper "L4: Practical loss-based stepsize adaptation for deep learning"☆124Updated 6 years ago
- Differentiable bitonic sorting☆141Updated 5 years ago
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆149Updated 8 years ago
- ☆135Updated 8 years ago
- Second-order optimiser for deep networks☆76Updated 7 years ago
- Code to reproduce some of the figures in the paper "On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima"☆146Updated 8 years ago
- Weight initialization schemes for PyTorch nn.Modules☆70Updated 8 years ago
- a lightweight and simple logger for Machine Learning☆126Updated 5 years ago
- Implementation and demonstration of backdrop in pytorch. Code and demonstration of GP dataset generator.☆68Updated 7 years ago
- Decoupled Neural Interfaces using Synthetic Gradients for PyTorch☆239Updated 7 years ago
- Python bindings for pyNVML and psutil library over network☆51Updated 2 years ago
- Implementation of Coulomb GANs☆62Updated 4 years ago
- Tools for loading standard data sets in machine learning☆206Updated 3 years ago
- Experiments with differentiable stacks and queues in PyTorch☆145Updated 6 years ago
- Evaluation code with models for the paper "On the Quantitative Analysis of Decoder-Based Generative Models"☆130Updated 8 years ago
- A Python implementation of the gradient REBAR estimator.☆46Updated 7 years ago
- PyTorch implementation of PathNet: Evolution Channels Gradient Descent in Super Neural Networks☆81Updated 8 years ago
- DeepArchitect: Automatically Designing and Training Deep Architectures☆145Updated 6 years ago
- repository for the Variational Autoencoder (VAE) blogpost series from Fast Forward Labs☆103Updated 8 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 6 years ago
- A Tensorfflow implementation of Attend, Infer, Repeat☆80Updated 7 years ago
- An extension to Sacred for automated hyperparameter optimization.☆59Updated 7 years ago
- a light weight experiment reproducibility toolset☆40Updated 5 years ago
- RL Experiments from our paper "Backpropagation Through the Void": https://arxiv.org/abs/1711.00123. Lovingly forked from OpenAI's RL Base…☆39Updated 7 years ago
- Pytorch implementation of DeepMind's differentiable neural computer paper.☆93Updated 8 years ago
- Implementations of polygamma, lgamma, and beta functions for PyTorch☆24Updated 8 years ago
- Code for "Deep Convolutional Networks as shallow Gaussian Processes"☆39Updated 6 years ago
- Benchmark and build RL architectures that can do multitask and transfer learning.☆144Updated 3 years ago