adityaiitb / pyprof2
PyProf2: PyTorch Profiling tool
☆82Updated 4 years ago
Alternatives and similar repositories for pyprof2:
Users that are interested in pyprof2 are comparing it to the libraries listed below
- ☆53Updated 7 years ago
- Programmable Neural Network Compression☆148Updated 2 years ago
- Simple Training and Deployment of Fast End-to-End Binary Networks☆158Updated 3 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- This is a PyTorch implementation of the Scalpel. Node pruning for five benchmark networks and SIMD-aware weight pruning for LeNet-300-100…☆41Updated 6 years ago
- MONeT framework for reducing memory consumption of DNN training☆174Updated 3 years ago
- DLPack for Tensorflow☆36Updated 4 years ago
- Customized matrix multiplication kernels☆53Updated 3 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 3 years ago
- Training neural networks with 8-bit computations☆28Updated 8 years ago
- Example repository for custom C++/CUDA operators for TorchScript☆115Updated 2 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated 2 months ago
- ☆109Updated 3 years ago
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆56Updated last year
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- Efficient DataLoader for PyTorch and Keras for loading datasets from web servers and object stores.☆29Updated 5 years ago
- ☆20Updated 7 years ago
- Efficient forward propagation for BCNNs☆50Updated 7 years ago
- Distributed, mixed-precision training with PyTorch☆89Updated 4 years ago
- Profile the GPU memory usage of every line in a Pytorch code☆82Updated 6 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- Training deep neural networks with low precision multiplications☆63Updated 9 years ago
- Train ImageNet in 18 minutes on AWS☆129Updated last year
- Example of backprop which uses constant memory☆43Updated 6 years ago
- Block-sparse primitives for PyTorch☆154Updated 3 years ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆124Updated 6 years ago
- This repository contains the results and code for the MLPerf™ Training v0.6 benchmark.☆42Updated last year