nki\_samples.reference.attention.flash\_attn\_bwd ================================================= .. currentmodule:: nki_samples.reference.attention .. autodata:: flash_attn_bwd