Santosh-Gupta / SpeedTorch
Library for faster pinned CPU <-> GPU transfer in Pytorch
☆683Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for SpeedTorch
- PyTorch layer-by-layer model profiler☆608Updated 3 years ago
- Deep Learning Experiment Management☆639Updated last year
- Implementations of ideas from recent papers☆391Updated 3 years ago
- Debug PyTorch code using PySnooper☆802Updated 3 years ago
- Experimental ground for optimizing memory of pytorch models☆360Updated 6 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆334Updated 5 years ago
- Fast Block Sparse Matrices for Pytorch☆545Updated 3 years ago
- torchbearer: A model fitting library for PyTorch☆636Updated 11 months ago
- 🔥 Reproducibly benchmarking Keras and PyTorch models☆367Updated 3 years ago
- PyTorch elastic training☆730Updated 2 years ago
- Sparse learning library and sparse momentum resources.☆379Updated 2 years ago
- Efficient Data Loading Pipeline in Pure Python☆210Updated 4 years ago
- TVM integration into PyTorch☆452Updated 4 years ago
- High-level batteries-included neural network training library for Pytorch☆398Updated 2 years ago
- Over9000 optimizer☆425Updated last year
- A GPU performance profiling tool for PyTorch models☆495Updated 3 years ago
- ☆532Updated 2 years ago
- Accelerate training by storing parameters in one contiguous chunk of memory.☆291Updated 4 years ago
- Tutorial for building a custom CUDA function for Pytorch☆514Updated 5 years ago
- Implementation of: "Exploring Randomly Wired Neural Networks for Image Recognition"☆685Updated 5 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆309Updated 4 years ago
- Estimates the size of a PyTorch model in memory☆352Updated 4 years ago
- Code for: "And the bit goes down: Revisiting the quantization of neural networks"☆636Updated 4 years ago
- Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …☆194Updated 4 years ago
- Official code for "Writing Distributed Applications with PyTorch", PyTorch Tutorial☆255Updated last year
- Efficient Learning of Augmentation Policy Schedules☆505Updated 5 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆369Updated 3 years ago
- A GPipe implementation in PyTorch☆818Updated 3 months ago
- Implementation for the Lookahead Optimizer.☆239Updated 2 years ago
- Easily evaluate machine learning models on public benchmarks☆171Updated 8 months ago