pytorch-extension
Here are 14 public repositories matching this topic...
Language:All
Sort:Most stars
Gradually-Warmup Learning Rate Scheduler for PyTorch
- Updated
Oct 10, 2024 - Python
Spatial Attentive Single-Image Deraining with a High Quality Real Rain Dataset (CVPR'19)
- Updated
Aug 30, 2021 - Jupyter Notebook
Polynomial Learning Rate Decay Scheduler for PyTorch
- Updated
Dec 25, 2021 - Python
A pyTorch Extension for Applied Mathematics
- Updated
Mar 17, 2020 - Python
Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops
- Updated
Mar 16, 2024 - Python
Template for CUDA / C++ extension writing with PyTorch
- Updated
Sep 9, 2020 - Python
Pytorch extension for openml-python
- Updated
Apr 26, 2025 - Jupyter Notebook
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
- Updated
Sep 11, 2021 - Python
XNOR-Net with binary conv2d kernels with XNOR GEMM op, support both CPU and GPU.
- Updated
Oct 25, 2022 - C
Binding C++ to PyTorch and extending PyTorch
- Updated
Nov 2, 2022 - Python
simple examples of tools and libraries
- Updated
Jul 31, 2025 - Python
PyTorch extension for alternative backward rules and gradient transforms (STE, gradient jamming, non-standard activations).
- Updated
Nov 30, 2025 - Python
Improve this page
Add a description, image, and links to thepytorch-extension topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with thepytorch-extension topic, visit your repo's landing page and select "manage topics."