pytorch / extension-cpp
C++ extensions in PyTorch
☆1,060Updated 3 weeks ago
Alternatives and similar repositories for extension-cpp:
Users that are interested in extension-cpp are comparing it to the libraries listed below
- ATen: A TENsor library for C++11☆691Updated 5 years ago
- PyTorch C++ API Documentation☆216Updated this week
- Tutorial for building a custom CUDA function for Pytorch☆513Updated 6 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,035Updated last year
- A GPipe implementation in PyTorch☆828Updated 6 months ago
- TVM integration into PyTorch☆452Updated 5 years ago
- A GPU performance profiling tool for PyTorch models☆501Updated 3 years ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,260Updated this week
- PyTorch layer-by-layer model profiler☆606Updated 3 years ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,677Updated this week
- Experimental ground for optimizing memory of pytorch models☆364Updated 6 years ago
- A lightweight library for PyTorch training tools and utilities☆1,685Updated this week
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆767Updated this week
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆904Updated this week
- common in-memory tensor structure☆942Updated last week
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 4 years ago
- C++ Implementation of PyTorch Tutorials for Everyone☆2,001Updated 9 months ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆596Updated 9 months ago
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆974Updated 5 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆594Updated 5 years ago
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,369Updated last year
- Dive into Deep Learning Compiler☆647Updated 2 years ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆696Updated 2 years ago
- Collective communications library with various primitives for multi-machine training.☆1,263Updated last week
- PyTorch extensions for high performance and large scale training.☆3,260Updated last month
- Fast Block Sparse Matrices for Pytorch☆546Updated 4 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,410Updated this week
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,499Updated 3 years ago