PyTorch half precision gemm lib w/ fused optional bias + optional relu/gelu
☆79Dec 3, 2024Updated last year
Alternatives and similar repositories for torch-cublas-hgemm
Users that are interested in torch-cublas-hgemm are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Flux diffusion model implementation using quantized fp8 matmul & remaining layers use faster half precision accumulate, which is ~2x fast…☆285Oct 12, 2024Updated last year
- Utilities for PyTorch distributed☆26Feb 27, 2025Updated last year
- [PACT'24] GraNNDis. A fast and unified distributed graph neural network (GNN) training framework for both full-batch (full-graph) and min…☆10Aug 13, 2024Updated 2 years ago
- A toolkit for scaling law research ⚖☆69Jan 27, 2025Updated last year
- ☆13Nov 27, 2025Updated 9 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops☆30Mar 16, 2024Updated 2 years ago
- ☆14Dec 22, 2024Updated last year
- ☆12Jan 4, 2024Updated 2 years ago
- JAX bindings for Flash Attention v2☆109Feb 28, 2026Updated 6 months ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- https://hf.co/hexgrad/Kokoro-82M☆14Jan 14, 2026Updated 8 months ago
- A CUDA kernel for NHWC GroupNorm for PyTorch☆26Nov 15, 2024Updated last year
- ☆18Mar 18, 2024Updated 2 years ago
- Supporting code for "LLMs for your iPhone: Whole-Tensor 4 Bit Quantization"☆11Mar 31, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆18Dec 2, 2024Updated last year
- FORA introduces simple yet effective caching mechanism in Diffusion Transformer Architecture for faster inference sampling.☆56Jul 8, 2024Updated 2 years ago
- Code for reproducibility of the E-swish paper experiments☆16Jan 23, 2018Updated 8 years ago
- ☆16Sep 30, 2023Updated 2 years ago
- ☆21Mar 3, 2025Updated last year
- Build modern UIs in Jupyter with Python☆12Dec 28, 2022Updated 3 years ago
- Code repo for "Model-Generated Pretraining Signals Improves Zero-Shot Generalization of Text-to-Text Transformers" (ACL 2023)☆22Nov 1, 2023Updated 2 years ago
- Graph model execution API for Candle☆18Jul 27, 2025Updated last year
- Code for my workshop "Production-ready WebAssembly with Rust" presented at RustLab 2023 in Florence☆15Nov 23, 2023Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆152Jun 16, 2024Updated 2 years ago
- ☆21Jun 6, 2024Updated 2 years ago
- A repo to do interpretability of pre-trained acoustic models☆15Oct 15, 2023Updated 2 years ago
- Waifu2x keras implementation https://github.com/nagadomi/waifu2x☆10Jun 19, 2016Updated 10 years ago
- Fast modular code to create and train cutting edge LLMs☆67May 16, 2024Updated 2 years ago
- ☆17Dec 12, 2021Updated 4 years ago
- trying to make WebGPU a bit easier to use☆19Jan 9, 2024Updated 2 years ago
- PyTorch implementation for OD-cheap-convolution.☆20Sep 29, 2019Updated 6 years ago
- ☆18Apr 3, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Triton-based implementation of Sparse Mixture of Experts.☆280Oct 3, 2025Updated 11 months ago
- RWKV6 in native pytorch and triton:)☆11Aug 4, 2024Updated 2 years ago
- Improved qwen image editing accuracy☆38Dec 2, 2025Updated 9 months ago
- Distributed IO-aware Attention algorithm☆24Sep 24, 2025Updated 11 months ago
- An upscaler node for flow-matching models like Qwen, applying the DemoFusion approach☆62Jan 29, 2026Updated 7 months ago
- A collection of utilities for FastHTML projects.☆14Oct 23, 2024Updated last year
- Extension for stable diffusion webui to add advance prompt tuning☆10Nov 13, 2022Updated 3 years ago