Support for flash-attn3

First of all many thanks for a cool and useful package!

I have one question about future directions and support. 
Are you planning on supporting flash-attention 3? 
I think this is only out in beta at the moment, but just wondering if the framework is in principle compatible? 
I could potentially try on H100/200 GPUs, so would be happy to help checking if this is something you are interested in. 