pytorch-extension
Here are 13 public repositories matching this topic...
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
Binding C++ to PyTorch and extending PyTorch
-
Updated
Nov 2, 2022 - Python
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
-
Updated
Sep 11, 2021 - Python
simple examples of tools and libraries
-
Updated
May 24, 2024 - Python
XNOR-Net with binary conv2d kernels with XNOR GEMM op, support both CPU and GPU.
-
Updated
Oct 25, 2022 - C
Pytorch extension for openml-python
-
Updated
Oct 21, 2024 - Jupyter Notebook
A pyTorch Extension for Applied Mathematics
-
Updated
Mar 17, 2020 - Python
Polynomial Learning Rate Decay Scheduler for PyTorch
-
Updated
Dec 25, 2021 - Python
Spatial Attentive Single-Image Deraining with a High Quality Real Rain Dataset (CVPR'19)
-
Updated
Aug 30, 2021 - Jupyter Notebook
Gradually-Warmup Learning Rate Scheduler for PyTorch
-
Updated
Oct 10, 2024 - Python
Improve this page
Add a description, image, and links to the pytorch-extension topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the pytorch-extension topic, visit your repo's landing page and select "manage topics."