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: about 2 years 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