aredden / torch-cublas-hgemmView external linksLinks
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
- ☆18Dec 2, 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 2 months ago
- A chat implementation for FastHTML☆11Sep 14, 2025Updated 5 months ago
- ☆13Nov 27, 2025Updated 2 months ago
- ☆12Jan 4, 2024Updated 2 years ago
- Utilities for PyTorch distributed☆25Feb 27, 2025Updated 11 months ago
- ☆18Mar 18, 2024Updated last year
- A CUDA kernel for NHWC GroupNorm for PyTorch☆22Nov 15, 2024Updated last year
- JAX bindings for Flash Attention v2☆103Feb 5, 2026Updated last week
- ☆19Dec 4, 2025Updated 2 months ago
- Repository for go shared libraries (for now).☆11Dec 1, 2025Updated 2 months ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated 8 months ago
- Supporting code for "LLMs for your iPhone: Whole-Tensor 4 Bit Quantization"☆11Mar 31, 2024Updated last year
- ☆21Mar 3, 2025Updated 11 months ago
- [GSI 2023] Learning Lagrangian Fluid Mechanics with E(3)-Equivariant GNNs☆15Jun 3, 2024Updated last year
- ☆19Apr 23, 2025Updated 9 months ago
- Extension for stable diffusion webui to add advance prompt tuning☆10Nov 13, 2022Updated 3 years ago
- ☆11Dec 22, 2024Updated last year
- ☆27Sep 7, 2025Updated 5 months ago
- Arrow Matrix Decomposition - Communication-Efficient Distributed Sparse Matrix Multiplication☆15Mar 25, 2024Updated last year
- A repo to do interpretability of pre-trained acoustic models☆15Oct 15, 2023Updated 2 years ago
- Cute layout visualization☆30Jan 18, 2026Updated 3 weeks ago
- Build modern UIs in Jupyter with Python☆12Dec 28, 2022Updated 3 years ago
- Small notebook to preprocess and evaluate images.☆14Nov 11, 2022Updated 3 years ago
- [ICLR 2025] Linear Combination of Saved Checkpoints Makes Consistency and Diffusion Models Better☆16Feb 15, 2025Updated last year
- ☆18Apr 3, 2023Updated 2 years ago
- RWKV6 in native pytorch and triton:)☆11Aug 4, 2024Updated last year
- TensorRT Extension for Stable Diffusion Web UI (Enhanced)☆13Feb 12, 2025Updated 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
- FORA introduces simple yet effective caching mechanism in Diffusion Transformer Architecture for faster inference sampling.☆52Jul 8, 2024Updated last year
- ☆50Mar 20, 2024Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆596Aug 12, 2025Updated 6 months ago
- JoyTag is a state of the art AI vision model for tagging images, with a focus on sex positivity and inclusivity. It uses the Danbooru tag…☆65May 22, 2024Updated last year
- ☆14Jan 10, 2024Updated 2 years ago