extension-cpp
by pytorch
C++ extensions in PyTorch
AI summary
C++ extension library
Enables the creation of custom C++ extensions with CUDA support in PyTorch
- stars
- 1K
- forks
- 215
- watching
- 35
Similar projects
Found by comparing what the projects do, not just their names.
PyTorch C++ implementation
A C++ implementation of PyTorch tutorials
C++ PyTorch inference toolkit
A repository providing tools and examples to serve PyTorch models as C++ inference applications.
CV library
Provides a PyTorch implementation of several computer vision tasks including object detection, segmentation and parsing.
Computer Vision Toolkit
A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.
PyTorch examples
A collection of code snippets and examples exploring various PyTorch concepts and techniques
Data loader library
Provides scalable, performant data loading solutions and utilities to be shared by PyTorch domain libraries
Model converter
Converts PyTorch models to their Torch equivalents.
Hadamard product layer
A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors.
Graph clustering library
A PyTorch extension library providing optimized graph cluster algorithms
Aligner
Implementation of RoIAlign and crop_and_resize functions for PyTorch
C++ compiler
A tool to import and compile C++ code from Python directly during runtime
Deep Learning Wrapper
Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages.
PyTorch utilities
A collection of utility functions to extend PyTorch functionality
ML toolkit
A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development.
