PyTorch half precision gemm lib w/ fused optional bias + optional relu/gelu
☆78Dec 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
Sorting:
- Flux diffusion model implementation using quantized fp8 matmul & remaining layers use faster half precision accumulate, which is ~2x fast…☆284Oct 12, 2024Updated last year
- A toolkit for scaling law research ⚖☆57Jan 27, 2025Updated last year
- Code repo for "Model-Generated Pretraining Signals Improves Zero-Shot Generalization of Text-to-Text Transformers" (ACL 2023)☆22Nov 1, 2023Updated 2 years ago
- Cuda extensions for PyTorch☆12Dec 2, 2025Updated 3 months ago
- A chat implementation for FastHTML☆11Sep 14, 2025Updated 5 months ago
- ☆13Nov 27, 2025Updated 3 months ago
- ☆12Jan 4, 2024Updated 2 years ago
- Utilities for PyTorch distributed☆25Feb 27, 2025Updated last year
- ☆18Dec 2, 2024Updated last year
- ☆18Mar 18, 2024Updated last year
- A CUDA kernel for NHWC GroupNorm for PyTorch☆23Nov 15, 2024Updated last year
- JAX bindings for Flash Attention v2☆103Feb 28, 2026Updated last week
- ☆19Dec 4, 2025Updated 3 months ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated 9 months ago
- Repository for go shared libraries (for now).☆11Dec 1, 2025Updated 3 months ago
- Supporting code for "LLMs for your iPhone: Whole-Tensor 4 Bit Quantization"☆11Mar 31, 2024Updated last year
- ☆22Jan 26, 2026Updated last month
- ☆21Mar 3, 2025Updated last year
- ☆19Apr 23, 2025Updated 10 months ago
- [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 last year
- ☆11Dec 22, 2024Updated last year
- Extension for stable diffusion webui to add advance prompt tuning☆10Nov 13, 2022Updated 3 years ago
- [GSI 2023] Learning Lagrangian Fluid Mechanics with E(3)-Equivariant GNNs☆15Jun 3, 2024Updated last year
- ☆27Sep 7, 2025Updated 6 months ago
- RWKV6 in native pytorch and triton:)☆11Aug 4, 2024Updated last year
- Cute layout visualization☆31Jan 18, 2026Updated last month
- [ICLR 2025] Linear Combination of Saved Checkpoints Makes Consistency and Diffusion Models Better☆16Feb 15, 2025Updated last year
- TensorRT Extension for Stable Diffusion Web UI (Enhanced)☆13Feb 12, 2025Updated last year
- Small notebook to preprocess and evaluate images.☆14Nov 11, 2022Updated 3 years ago
- ☆15Jun 19, 2024Updated last year
- ☆18Apr 3, 2023Updated 2 years ago
- Build modern UIs in Jupyter with Python☆12Dec 28, 2022Updated 3 years ago
- A repo to do interpretability of pre-trained acoustic models☆15Oct 15, 2023Updated 2 years ago
- ☆13Sep 30, 2023Updated 2 years ago
- Arrow Matrix Decomposition - Communication-Efficient Distributed Sparse Matrix Multiplication☆15Mar 25, 2024Updated last year
- Utilities for Training Very Large Models☆58Sep 25, 2024Updated last year
- Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops☆30Mar 16, 2024Updated last year
- ☆11Jun 13, 2025Updated 8 months ago
- Simple implementation of a GPT (training and inference) in PyTorch.☆13Dec 11, 2023Updated 2 years ago