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
215 forks
Language: Python
last commit: 7 months ago Related projects:
| Repository | Description | Stars |
|---|---|---|
 prabhuomkar/pytorch-cpp | A C++ implementation of PyTorch tutorials | 1,978 |
 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,191 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,236 |
 ptrblck/pytorch_misc | A collection of code snippets and examples exploring various PyTorch concepts and techniques | 547 |
| pytorch/data | Provides scalable, performant data loading solutions and utilities to be shared by PyTorch domain libraries | 1,149 |
 clcarwin/convert_torch_to_pytorch | Converts PyTorch models to their Torch equivalents. | 541 |
 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 | 838 |
 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,195 |
 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,523 |