https://lightning.ai/pages/community/tutorial/accelerating-large-language-models-with-mixed-precision-techniques/