nki\_samples.reference.attention.flash\_fwd =========================================== .. currentmodule:: nki_samples.reference.attention .. autodata:: flash_fwd