https://medium.com/@bnjmn_marie/estimate-the-memory-consumption-of-llms-for-inference-and-fine-tuning-c28c8b9e3274