Release a new version of flash attention with causal prefix cross-attention.

• Since only the forward pass is implemented, this should be released on its own branch. We should maybe also block backprop from running or something...
• Include tests