Add flash_attn_with_kvcache paged-decode kernel (port from huggingface/transformers#45977)
#4 opened 11 days ago
by 
ArthurZ
Add flash_attn_func + harden MPS dispatch for transformers compatibility
#3 opened 14 days ago
by ArthurZ
Test code with transformers
2
#1 opened 7 months ago
by 
manueldeprada