pytorch / extension-scriptLinks
Example repository for custom C++/CUDA operators for TorchScript
☆114Updated 2 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 4 years ago
- Codebase associated with the PyTorch compiler tutorial☆45Updated 5 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 5 years ago
- ☆53Updated 7 years ago
- convert torch module to tensorrt network or tvm function☆88Updated 5 years ago
- Serving PyTorch 1.0 Models as a Web Server in C++☆226Updated 5 years ago
- COCO API - Dataset @ http://cocodataset.org/☆54Updated 8 months ago
- Efficient Data Loading Pipeline in Pure Python☆212Updated 4 years ago
- Benchmark of TVM quantized model on CUDA☆111Updated 4 years ago
- Model Scope in PyTorch (include Params, FLOPs, Madds).☆122Updated 6 years ago
- PyTorch 1.0 inference in C++ on Windows10 platforms☆89Updated 6 years ago
- TVM integration into PyTorch☆452Updated 5 years ago
- Use TensorFlow efficiently☆95Updated 4 years ago
- Swish Activation - PyTorch CUDA Implementation☆37Updated 5 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆145Updated 3 years ago
- Simple Training and Deployment of Fast End-to-End Binary Networks☆157Updated 3 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆121Updated 6 years ago
- ☆45Updated 2 years ago
- Official code for "Writing Distributed Applications with PyTorch", PyTorch Tutorial☆262Updated 2 years ago
- thxx: libtorch C++ API extentions and examples☆45Updated 6 years ago
- Fast CUDA Kernels for ResNet Inference.☆174Updated 6 years ago
- Scailable ONNX python tools☆97Updated 7 months ago
- ☆69Updated 5 years ago
- Some recent Quantizing techniques on PyTorch☆72Updated 5 years ago
- Inference of quantization aware trained networks using TensorRT☆81Updated 2 years ago
- Network acceleration methods☆177Updated 3 years ago
- PyTorch C++ API Documentation☆228Updated 2 weeks ago
- Personal collection of references for high performance mixed precision training.☆41Updated 5 years ago
- pytorch model summary, statistic parameters number, memory usage, FLOPs and so on☆98Updated 5 years ago
- Mobilenet model converted from tensorflow☆48Updated 6 years ago