pytorch-extension

There are 0 repository under pytorch-extension topic.

ildoonet / pytorch-gradual-warmup-lr
Gradually-Warmup Learning Rate Scheduler for PyTorch
deep-learning large-scale-learning learning-rate-decay multigpu multinode pytorch pytorch-extension
Language:Python 967
stevewongv / SPANet
Spatial Attentive Single-Image Deraining with a High Quality Real Rain Dataset (CVPR'19)
cvpr2019 computer-vision low-level-vision deraining pytorch cuda cupy pytorch-extension
Language:Jupyter Notebook 200
stevewongv / DSC-PyTorch
A PyTorch implementation of "Direction-Aware Spatial Context Features for Shadow Detection" CVPR'18 | T-PAMI'19
pytorch pytorch-extension dsc cvpr2018 pami cvpr
Language:Python 90
cmpark0126 / pytorch-polynomial-lr-decay
Polynomial Learning Rate Decay Scheduler for PyTorch
pytorch deep-learning learning-rate-decay learning-rate-scheduling pytorch-extension
Language:Python 65
ZichaoLong / aTEAM
A pyTorch Extension for Applied Mathematics
pytorch-extension applied-mathematics
Language:Python 38
aredden / torch-bnb-fp4
Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops
cuda-kernels python pytorch pytorch-extension quantization
Language:Python 17
artitw / apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
pytorch pytorch-extension apex amp
Language:Python 5
frgfm / torch-cuda-template
Template for CUDA / C++ extension writing with PyTorch
cpp cuda pytorch pytorch-extension
Language:Python 5
pminhtam / xnor_conv_pytorch_extension
XNOR-Net with binary conv2d kernels with XNOR GEMM op, support both CPU and GPU.
binary-convolutions binary-op cpp cuda pytorch-extension xnor-convolutions xnor-net binary-neural-networks pytorch gemm
Language:C 4
openml / openml-pytorch
Pytorch extension for openml-python
openml pytorch pytorch-extension python hacktoberfest
Language:Python 3
AmirMardan / pytorch_extending_cpp_binding
Binding C++ to PyTorch and extending PyTorch
autograd cpp-bindings pytorch-examples pytorch-extension pytorch-learning torch-cpp
Language:Python 1
djalex88 / gaussian_rbf
PyTorch extension
pytorch-extension pytorch machine-learning
Language:Cuda 1
eshibusawa / Simple-Examples
simple examples of tools and libraries
cub cuda cupy pybind11 python pytorch-extension tensorcore
Language:Python 0