Summary: Optimum Neuron serves as the bridge between Hugging Face libraries, such as Transformers, Diffusers, and PEFT, and AWS Trainium and Inferentia accelerators. It provides a set of tools enabling easy model loading, training, and inference on both single and multiple Neuron core configurations, across a wide range of downstream tasks.
Author: "HuggingFace Inc. Special Ops Team" <hardware@huggingface.co>
Weekly downloads over the last yearSeptemberOctoberNovemberDecember2026FebruaryMarchAprilMayJuneJulyAugustDate00.511.522.533.54 thousand downloads per week
Source: ClickPy
Dependencies
Optimum-neuron has 43 dependencies, 40 of which optional.