adityaiitb / pyprof2
PyProf2: PyTorch Profiling tool
☆83Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for pyprof2
- ☆53Updated 6 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
- Programmable Neural Network Compression☆147Updated 2 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 2 years ago
- Codebase associated with the PyTorch compiler tutorial☆44Updated 5 years ago
- DLPack for Tensorflow☆36Updated 4 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 3 years ago
- Customized matrix multiplication kernels☆53Updated 2 years ago
- PyTorch implementation of L2L execution algorithm☆106Updated last year
- Simple Training and Deployment of Fast End-to-End Binary Networks☆159Updated 2 years ago
- ☆107Updated 3 years ago
- Profile the GPU memory usage of every line in a Pytorch code☆82Updated 6 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆129Updated 2 years ago
- Block-sparse primitives for PyTorch☆148Updated 3 years ago
- Online Normalization for Training Neural Networks (Companion Repository)☆79Updated 3 years ago
- Training neural networks with 8-bit computations☆29Updated 8 years ago
- MONeT framework for reducing memory consumption of DNN training☆173Updated 3 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆117Updated 8 months ago
- Make TFRecord Usable Again☆87Updated last year
- ☆47Updated 4 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- Use TensorFlow efficiently☆95Updated 3 years ago
- Simple implementation of the LSUV initialization in PyTorch☆58Updated 10 months ago
- Distributed, mixed-precision training with PyTorch☆89Updated 4 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 4 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆173Updated 5 years ago
- Repository containing pruned models and related information☆36Updated 3 years ago
- A Re-implementation of Fixed-update Initialization☆151Updated 5 years ago
- PyTorch DataLoader processed in multiple remote computation machines for heavy data processings☆66Updated 5 years ago