https://huggingface.co/docs/transformers/performance#anatomy-of-models-operations for example  cc @mishig25