chrischoy / MakePytorchPlusPlus
How and why you want to make your pytorch CUDA/CPP extension with a Makefile
☆173Updated 5 years ago
Related projects ⓘ
Alternatives and complementary repositories for MakePytorchPlusPlus
- Tutorial for building a custom CUDA function for Pytorch☆514Updated 5 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆117Updated 8 months ago
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 2 years ago
- CuPy fused PyTorch neural networks ops☆274Updated 6 years ago
- Pytorch Custom CUDA kernel for searchsorted☆135Updated last year
- Distributed, mixed-precision training with PyTorch☆89Updated 4 years ago
- Programmable Neural Network Compression☆147Updated 2 years ago
- ☆165Updated 5 years ago
- A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.☆238Updated 3 years ago
- PyTorch layer-by-layer model profiler☆607Updated 3 years ago
- Experimental ground for optimizing memory of pytorch models☆360Updated 6 years ago
- Repository has been moved: https://github.com/adobe/antialiased-cnns☆165Updated 3 years ago
- A Pytorch implementation of "LegoNet: Efficient Convolutional Neural Networks with Lego Filters" (ICML 2019).☆140Updated 4 years ago
- PyProf2: PyTorch Profiling tool☆83Updated 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
- Efficient Data Loading Pipeline in Pure Python☆210Updated 4 years ago
- Official code for "Writing Distributed Applications with PyTorch", PyTorch Tutorial☆255Updated last year
- On Network Design Spaces for Visual Recognition☆94Updated 4 years ago
- Examples of C extensions for PyTorch☆258Updated last year
- Model Scope in PyTorch (include Params, FLOPs, Madds).☆121Updated 5 years ago
- A Re-implementation of Fixed-update Initialization☆151Updated 5 years ago
- Utilities for Pytorch☆89Updated 2 years ago
- An example of C++ extension for PyTorch.☆36Updated 5 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆122Updated 5 years ago
- Sparse Blocks Networks☆436Updated 6 years ago
- ☆61Updated 4 years ago
- An asynchronous pytorch Dataloader for general neural network pipeline accelaration.☆53Updated 10 months ago
- Block-sparse primitives for PyTorch☆148Updated 3 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- ☆180Updated last year