The official repo of Qwen (通义千问) chat & pretrained large language model proposed by Alibaba Cloud.
-
Updated
Mar 5, 2026 - Python
#
flash-attention
Here are 55 public repositories matching this topic...
Official release of InternLM series (InternLM, InternLM2, InternLM2.5, InternLM3).
chatbot chinese gpt pretrained-models llm long-context rlhf large-language-model flash-attention fine-tuning-llm
-
Updated
Oct 30, 2025 - Python
📚A curated list of Awesome LLM/VLM Inference Papers with Codes: Flash-Attention, Paged-Attention, WINT8/4, Parallelism, etc.🎉
mla vllm llm-inference awesome-llm flash-attention tensorrt-llm paged-attention deepseek flash-attention-3 deepseek-v3 minimax-01 deepseek-r1 flash-mla qwen3
-
Updated
Mar 26, 2026 - Python
MoBA: Mixture of Block Attention for Long-Context LLMs
-
Updated
Apr 3, 2025 - Python
InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencies.
pytorch multi-modal gemma pipeline-parallelism transformers-models tensor-parallelism llava llm-training internlm flash-attention zero3 llm-framework sequence-parallelism internlm2 ring-attention deepspeed-ulysses llama3 910b
-
Updated
Aug 21, 2025 - Python
Trainable fast and memory-efficient sparse attention
-
Updated
Mar 27, 2026 - Python
[ICLR 2026] rCM: SOTA JVP-Based Diffusion Distillation & Few-Step Video Generation & Scaling Up sCM/MeanFlow
real-time diffusion distillation video-generation world-models distribution-matching jacobian-vector-product flash-attention consistency-model wan-video meanflow few-step-generation autoregressive-video-generation
-
Updated
Mar 24, 2026 - Python
FlashSinkhorn: IO-Aware Entropic Optimal Transport in PyTorch + Triton. Streaming Sinkhorn with O(nd) memory.
machine-learning gpu cuda pytorch triton optimal-transport sinkhorn flash-attention entropic-optimal-transport flashsinkhorn
-
Updated
Mar 3, 2026 - Python
Triton implementation of FlashAttention2 that adds Custom Masks.
deep-learning triton attention cuda-kernels attention-mechanism triton-lang flash-attention flash-attention-2
-
Updated
Aug 14, 2024 - Python
[CVPR 2025 Highlight] The official CLIP training codebase of Inf-CL: "Breaking the Memory Barrier: Near Infinite Batch Size Scaling for Contrastive Loss". A super memory-efficiency CLIP training scheme.
-
Updated
Jan 16, 2025 - Python
Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.
nlp bloom pipeline pytorch deepspeed llm full-finetune model-parallization flash-attention llama2 baichuan2-7b chatglm3-6b mixtral-8x7b
-
Updated
Feb 5, 2024 - Python
Automatically benchmark and optimize attention in diffusion models. 1.5-2x speedup on RTX 4090.
flux performance optimization attention diffusion stable-diffusion comfyui flash-attention comfyui-custom-node sageattention
-
Updated
Feb 9, 2026 - Python
Python package for rematerialization-aware gradient checkpointing
-
Updated
Oct 31, 2023 - Python
Pre-built wheels that erase Flash Attention 3 installation headaches.
-
Updated
Mar 31, 2026 - Python
Utilities for efficient fine-tuning, inference and evaluation of code generation models
-
Updated
Oct 3, 2023 - Python
Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.
-
Updated
Nov 4, 2024 - Python
A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/Pallas/JAX).
-
Updated
Mar 4, 2025 - Python
Long term project about a custom AI architecture. Consist of cutting-edge technique in machine learning such as Flash-Attention, Group-Query-Attention, ZeRO-Infinity, BitNet, etc.
python machine-learning deep-learning jupyter-notebook pytorch transformer bitnet pytorch-lightning vision-transformer large-language-models low-rank-adaptation flash-attention
-
Updated
Oct 15, 2024 - Python
Toy Flash Attention implementation in torch
-
Updated
Sep 22, 2024 - Python
Improve this page
Add a description, image, and links to the flash-attention topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the flash-attention topic, visit your repo's landing page and select "manage topics."