GitHub topics: pytorch-extension
eshibusawa/Simple-Examples
simple examples of tools and libraries
Language: Python - Size: 91.8 KB - Last synced at: 14 days ago - Pushed at: 14 days ago - Stars: 0 - Forks: 0

ildoonet/pytorch-gradual-warmup-lr
Gradually-Warmup Learning Rate Scheduler for PyTorch
Language: Python - Size: 52.7 KB - Last synced at: 24 days ago - Pushed at: 10 months ago - Stars: 990 - Forks: 127

openml/openml-pytorch
Pytorch extension for openml-python
Language: Jupyter Notebook - Size: 10.6 MB - Last synced at: 17 days ago - Pushed at: 3 months ago - Stars: 6 - Forks: 0

cmpark0126/pytorch-polynomial-lr-decay
Polynomial Learning Rate Decay Scheduler for PyTorch
Language: Python - Size: 10.7 KB - Last synced at: 3 months ago - Pushed at: over 3 years ago - Stars: 65 - Forks: 13

stevewongv/DSC-PyTorch
A PyTorch implementation of "Direction-Aware Spatial Context Features for Shadow Detection" CVPR'18 | T-PAMI'19
Language: Python - Size: 46.9 KB - Last synced at: 11 months ago - Pushed at: 11 months ago - Stars: 90 - Forks: 30

frgfm/torch-cuda-template
Template for CUDA / C++ extension writing with PyTorch
Language: Python - Size: 14.6 KB - Last synced at: 4 months ago - Pushed at: almost 5 years ago - Stars: 6 - Forks: 0

AmirMardan/pytorch_extending_cpp_binding
Binding C++ to PyTorch and extending PyTorch
Language: Python - Size: 5.86 KB - Last synced at: about 1 year ago - Pushed at: over 2 years ago - Stars: 1 - Forks: 0

aredden/torch-bnb-fp4
Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops
Language: Python - Size: 72.3 KB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 13 - Forks: 0

stevewongv/SPANet
Spatial Attentive Single-Image Deraining with a High Quality Real Rain Dataset (CVPR'19)
Language: Jupyter Notebook - Size: 3.04 MB - Last synced at: almost 2 years ago - Pushed at: almost 4 years ago - Stars: 191 - Forks: 54

ZichaoLong/aTEAM
A pyTorch Extension for Applied Mathematics
Language: Python - Size: 78.1 KB - Last synced at: almost 2 years ago - Pushed at: over 5 years ago - Stars: 35 - Forks: 8

djalex88/gaussian_rbf
PyTorch extension
Language: Cuda - Size: 12.7 KB - Last synced at: over 2 years ago - Pushed at: about 4 years ago - Stars: 1 - Forks: 0

pminhtam/xnor_conv_pytorch_extension
XNOR-Net with binary conv2d kernels with XNOR GEMM op, support both CPU and GPU.
Language: C - Size: 66.4 KB - Last synced at: over 2 years ago - Pushed at: almost 3 years ago - Stars: 1 - Forks: 0

artitw/apex Fork of NVIDIA/apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Language: Python - Size: 13.8 MB - Last synced at: 23 days ago - Pushed at: almost 4 years ago - Stars: 5 - Forks: 0
