Understanding the FlashAttention papers and the various implementations.
Choices to keep in mind while adding mlflow logging to a codebase