Package profile
optimum-neuron
- 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>
- Homepage: https://huggingface.co/docs/optimum-neuron/index
- Number of releases: 34
- First release: 0.0.1 on 2023-03-13
- Latest release: 0.3.0 on 2025-07-18
Releases
PyPI Downloads
Loading PyPI statistics...
Dependencies
Optimum-neuron has 45 dependencies, 39 of which optional.| Dependency | Optional |
|---|---|
| accelerate | false |
| huggingface-hub | false |
| numpy | false |
| optimum | false |
| protobuf | false |
| transformers | false |
| cloudpickle | true |
| 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-order | 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 |
| torchcodec | true |
| torchvision | true |
| trl | true |
| vllm | true |
| wheel | true |