extension-cpp
C++ extension library
Enables the creation of custom C++ extensions with CUDA support in PyTorch
C++ extensions in PyTorch
1k stars
35 watching
213 forks
Language: Python
last commit: 4 months ago Related projects:
| Repository | Description | Stars |
|---|---|---|
 prabhuomkar/pytorch-cpp | A C++ implementation of PyTorch tutorials | 1,965 |
 wizaron/pytorch-cpp-inference | A repository providing tools and examples to serve PyTorch models as C++ inference applications. | 226 |
 openseg-group/openseg.pytorch | Provides a PyTorch implementation of several computer vision tasks including object detection, segmentation and parsing. | 1,190 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,217 |
 ptrblck/pytorch_misc | A collection of code snippets and examples exploring various PyTorch concepts and techniques | 544 |
| pytorch/data | A PyTorch project providing data loading utilities and scalable dataloading solutions | 1,133 |
 clcarwin/convert_torch_to_pytorch | Converts PyTorch models to their Torch equivalents. | 539 |
 sniklaus/pytorch-extension | A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors. | 117 |
 rusty1s/pytorch_cluster | A PyTorch extension library providing optimized graph cluster algorithms | 824 |
 pollardbanknote/cppbackport | A C++ library that ported new features to older standards | 38 |
 longcw/roialign.pytorch | Implementation of RoIAlign and crop_and_resize functions for PyTorch | 554 |
 tbenthompson/cppimport | A tool to import and compile C++ code from Python directly during runtime | 1,187 |
 hughperkins/pytorch | Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. | 432 |
 mrdrozdov-github/pytorch-extras | A collection of utility functions to extend PyTorch functionality | 50 |
 bloodaxe/pytorch-toolbelt | A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development. | 1,520 |