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
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- ☆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
- Simple Training and Deployment of Fast End-to-End Binary Networks☆158Updated 2 years ago
- DLPack for Tensorflow☆36Updated 4 years ago
- Training neural networks with 8-bit computations☆29Updated 8 years ago
- Customized matrix multiplication kernels☆53Updated 2 years ago
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 2 years ago
- Programmable Neural Network Compression☆148Updated 2 years ago
- Repository containing pruned models and related information☆37Updated 3 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆124Updated 6 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 3 years ago
- Codebase associated with the PyTorch compiler tutorial☆44Updated 5 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- Block-sparse primitives for PyTorch☆154Updated 3 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 2 years ago
- ☆26Updated 2 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 4 years ago
- ☆109Updated 3 years ago
- A PyTorch implementation of "Incremental Network Quantization: Towards Lossless CNNs with Low-Precision Weights"☆166Updated 4 years ago
- Training with FP16 weights in PyTorch☆77Updated 5 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- MONeT framework for reducing memory consumption of DNN training☆174Updated 3 years ago
- custom cuda kernel for {2, 3}d relative attention with pytorch wrapper☆43Updated 4 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated last week
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 5 years ago
- Use TensorFlow efficiently☆95Updated 3 years ago
- Training wide residual networks for deployment using a single bit for each weight☆36Updated 4 years ago