[pyrophane]

Here is a good reference:

https://huggingface.co/docs/transformers/perf_train_gpu_one