Package profile
optimum-neuron
- Summary: Optimum Neuron is the interface between the Hugging Face Transformers and Diffusers libraries and AWS Trainium and Inferentia accelerators. It provides a set of tools enabling easy model loading, training and inference on single and multiple neuron core settings for different downstream tasks.
- Author: HuggingFace Inc. Special Ops Team
- Homepage: https://huggingface.co/hardware/aws
- Number of releases: 33
- First release: 0.0.1 on 2023-03-13
- Latest release: 0.2.2 on 2025-07-01
Releases
PyPI Downloads
Dependencies
Optimum-neuron has 41 dependencies, 35 of which optional.| Dependency | Optional |
|---|---|
| accelerate | false |
| huggingface-hub | false |
| numpy | false |
| optimum | false |
| protobuf | false |
| transformers | false |
| compel | true |
| controlnet-aux | true |
| datasets | true |
| diffusers | true |
| evaluate | true |
| GitPython | true |
| hf-transfer | true |
| isort | true |
| libneuronxla | true |
| librosa | true |
| mediapipe | true |
| neuron-cc | true |
| neuronx-cc | true |
| neuronx-distributed | true |
| opencv-python-headless | true |
| parameterized | true |
| peft | true |
| psutil | true |
| pytest | true |
| pytest-rerunfailures | true |
| pytest-timeout | true |
| rjieba | true |
| ruff | true |
| sacremoses | true |
| safetensors | true |
| sentence-transformers | true |
| sentencepiece | true |
| soundfile | true |
| timm | true |
| torch | true |
| torch-neuron | true |
| torch-neuronx | true |
| torchvision | true |
| trl | true |
| wheel | true |