Optimum

Hugging Face Optimum is a powerful extension of Transformers and Diffusers that provides optimization tools for efficient model training and runtime on targeted hardware.

GitHub