Gradually-Warmup Learning Rate Scheduler for PyTorch
-
Updated
Oct 10, 2024 - Python
#
pytorch-extension
Here are 19 public repositories matching this topic...
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
Minimal FlashAttention in CUDA C++/CuTe: readable WMMA/CuTe kernels, no NxN workspace, up to 4.5x faster than naive PyTorch
cuda attention cutlass cute gpu-kernels pytorch-extension tensor-cores llm flash-attention flashattention wmma
-
Updated
Jun 2, 2026 - Cuda
Pytorch extension for openml-python
-
Updated
May 19, 2026 - Jupyter Notebook
Template for CUDA / C++ extension writing with PyTorch
-
Updated
Sep 9, 2020 - Python
XNOR-Net with binary conv2d kernels with XNOR GEMM op, support both CPU and GPU.
cpp cuda pytorch xnor-net gemm binary-convolutions xnor-convolutions binary-neural-networks binary-op pytorch-extension
-
Updated
Oct 25, 2022 - C
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
-
Updated
Sep 11, 2021 - Python
Pre-compiled custom CUDA extension for Block Sparse Attention (Python 3.11 / PyTorch 2.6.0+cu124).
python machine-learning cuda pytorch pytorch-extension llm-optimization cuda-12 block-sparse-attention precompiled-wheel
-
Updated
May 4, 2026
Binding C++ to PyTorch and extending PyTorch
-
Updated
Nov 2, 2022 - Python
The numerical continuity layer for GPU computing
-
Updated
May 6, 2026 - Python
PyTorch extension for alternative backward rules and gradient transforms (STE, gradient jamming, non-standard activations).
training research deep-learning python3 pytorch autograd neural-networks gradient backpropagation binary-neural-networks activation-functions gradient-clipping pytorch-extension gradient-modification
-
Updated
Nov 30, 2025 - Python
simple examples of tools and libraries
-
Updated
Jan 27, 2026 - Python
High-performance matrix engine for Unit-Domain Flow (UDF). Eliminates Mantissa Friction with 0.00 MSE integrity.
python ai deep-learning neural-network linear-algebra cuda pytorch artificial-intelligence matrix-multiplication neural-networks low-latency gpu-computing math-library performance-optimization pytorch-extension matrix-multiplication-parallel tensor-engine physics-informed-neural-networks ai-infrastructure cuda-optimization
-
Updated
Feb 17, 2026 - Python
CUDA kernels for LLM decode-stage inference, built as a PyTorch extension with correctness tests and latency benchmarks.
-
Updated
May 21, 2026 - 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."