huggingface/optimum
๐ Accelerate inference and training of ๐ค Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools
๐ A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
๐ Accelerate inference and training of ๐ค Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Tensorforce: a TensorFlow library for applied reinforcement learning
On-device AI across mobile, embedded and edge for PyTorch
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Machine learning metrics for distributed, scalable PyTorch applications.
1 capture since 2026-05-25