pytorch / extension-scriptLinks
Example repository for custom C++/CUDA operators for TorchScript
☆114Updated 3 years ago
Alternatives and similar repositories for extension-script
Users that are interested in extension-script are comparing it to the libraries listed below
Sorting:
- PyProf2: PyTorch Profiling tool☆82Updated 5 years ago
- Serving PyTorch 1.0 Models as a Web Server in C++☆226Updated 5 years ago
- PyTorch 1.0 inference in C++ on Windows10 platforms☆89Updated 6 years ago
- Efficient Data Loading Pipeline in Pure Python☆212Updated 5 years ago
- ☆54Updated 7 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 6 years ago
- Code for https://arxiv.org/abs/1810.04622☆140Updated 6 years ago
- This is an example of how to train a MNIST network in Python and run it in c++ with pytorch 1.0☆96Updated 6 years ago
- Personal collection of references for high performance mixed precision training.☆41Updated 5 years ago
- Use TensorFlow efficiently☆95Updated 4 years ago
- Model Scope in PyTorch (include Params, FLOPs, Madds).☆122Updated 6 years ago
- Training with FP16 weights in PyTorch☆80Updated 6 years ago
- Simple Training and Deployment of Fast End-to-End Binary Networks☆158Updated 3 years ago
- convert torch module to tensorrt network or tvm function☆89Updated 5 years ago
- ☆108Updated 4 years ago
- Codebase associated with the PyTorch compiler tutorial☆46Updated 6 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆122Updated 6 years ago
- MnasNet snapshot☆35Updated 6 years ago
- ☆26Updated 5 years ago
- ☆33Updated last year
- Efficient DataLoader for PyTorch and Keras for loading datasets from web servers and object stores.☆30Updated 5 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 5 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆146Updated 4 years ago
- Programmable Neural Network Compression☆150Updated 3 years ago
- PyTorch C++ API Documentation☆237Updated this week
- A PyTorch implementation of "Incremental Network Quantization: Towards Lossless CNNs with Low-Precision Weights"☆168Updated 5 years ago
- Prunable nn layers for pytorch.☆48Updated 7 years ago
- Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …☆198Updated 5 years ago
- A Pytorch implementation of Neural Network Compression (pruning, deep compression, channel pruning)☆156Updated last year
- Benchmark of TVM quantized model on CUDA☆111Updated 5 years ago