Gradually-Warmup Learning Rate Scheduler for PyTorch
-
Updated
Oct 10, 2024 - Python
#
pytorch-extension
Here are 16 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
Pytorch extension for openml-python
-
Updated
Apr 14, 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
Binding C++ to PyTorch and extending PyTorch
-
Updated
Nov 2, 2022 - Python
⚡ LLM-Speed: High-performance CUDA kernels for LLM inference — FlashAttention with O(N) memory, Tensor Core GEMM (95% cuBLAS), and seamless PyTorch integration. Supports Volta to Hopper GPUs.
deep-learning cuda matrix-multiplication gpu-acceleration attention cuda-kernels gpu-computing memory-efficient attention-mechanism gemm pybind11 pytorch-extension inference-optimization cuda-programming llm tensor-core llm-inference flashattention kernel-optimization flashattention-2
-
Updated
Apr 21, 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
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."