Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.
☆16Aug 31, 2023Updated 2 years ago
Alternatives and similar repositories for flash_attention_inference
Users that are interested in flash_attention_inference are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Anatomy of High-Performance GEMM with Online Fault Tolerance on GPUs☆14Apr 3, 2025Updated last year
- Standalone Flash Attention v2 kernel without libtorch dependency☆113Sep 10, 2024Updated last year
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆45Feb 27, 2025Updated last year
- CPU Memory Compiler and Parallel programing☆26Nov 18, 2024Updated last year
- GEMV implementation with CUTLASS☆21Aug 21, 2025Updated 9 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Implement Flash Attention using Cute.☆108Dec 17, 2024Updated last year
- Aligntune : A Modular Toolkit for Post Training Alignment of LLMs☆37Updated this week
- ☆13Dec 22, 2024Updated last year
- High-performance GEMM implementation optimized for NVIDIA H100 GPUs, leveraging Hopper architecture's TMA, WGMMA, and Thread Block Cluste…☆11Dec 4, 2024Updated last year
- This project is about convolution operator optimization on GPU, include GEMM based (Implicit GEMM) convolution.☆44Sep 29, 2025Updated 8 months ago
- Vstream - Video Analytics pipeline with Hardware based accelerations (dev - stage)☆10Feb 2, 2024Updated 2 years ago
- Flash Attention in raw Cuda C beating PyTorch☆39May 14, 2024Updated 2 years ago
- transformer tokenizers (e.g. BERT tokenizer) in C++ (WIP)☆18Apr 7, 2022Updated 4 years ago
- Flash Attention in ~100 lines of CUDA (forward pass only)☆12Jun 10, 2024Updated 2 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆15Oct 9, 2022Updated 3 years ago
- Fast SGEMM emulation on Tensor Cores☆17Feb 16, 2025Updated last year
- Fast and low-memory attention layer written in CUDA☆20Jul 14, 2023Updated 2 years ago
- 2014: Variational Monte Carlo for the harmonic oscillator, helium, hydrogen and H2 - IPython notebook and FORTRAN90☆13Jun 23, 2016Updated 9 years ago
- ☆66Feb 15, 2026Updated 4 months ago
- ☆160Dec 26, 2024Updated last year
- Estimate depth from surface normal.☆12Aug 14, 2020Updated 5 years ago
- An object tracking project with YOLOv5-v5.0 and Deepsort, speed up by C++ and TensorRT.☆16Oct 23, 2025Updated 7 months ago
- High Performance FP8 GEMM Kernels for SM89 and later GPUs.☆21Jan 24, 2025Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- A tiny FP8 multiplication unit written in Verilog. TinyTapeout 2 submission.☆14Nov 23, 2022Updated 3 years ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆47Jun 11, 2025Updated last year
- Deep insight tensorrt, including but not limited to qat, ptq, plugin, triton_inference, cuda☆23Jun 10, 2026Updated last week
- Stochastic Series Expansion (SSE) for a isotropic S=1/2 antiferromagnetic quantum Heisenberg model in 1D, 2D or 3D lattice . Every lattic…☆15Jan 23, 2021Updated 5 years ago
- ☆16Jun 11, 2026Updated last week
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆81Aug 12, 2024Updated last year
- learning notes when learning the source code of pytorch☆24Apr 3, 2019Updated 7 years ago
- Retargetable ML compilers for the twenty-first century!☆13Apr 22, 2025Updated last year
- IREE compiler and runtime for Snitch☆15May 14, 2026Updated last month
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- CUDA SGEMM optimization note☆15Oct 31, 2023Updated 2 years ago
- PointPillars TensorRT version pretrained on MMDetection3d with WaymoOpenDataset☆23Aug 11, 2022Updated 3 years ago
- Effective transpose on Hopper GPU☆29Sep 6, 2025Updated 9 months ago
- Llama causal LM fully recreated in LibTorch. Designed to be used in Unreal Engine 5☆16Sep 19, 2024Updated last year
- Code for training & inference with FLAN family of models☆17May 23, 2023Updated 3 years ago
- High Performance Int8 GEMM Kernels for SM80 and later GPUs.☆23Mar 11, 2025Updated last year
- A fork of the BEVDet series .☆22Oct 8, 2023Updated 2 years ago