PyPI page
Home page
Author:
HuggingFace Inc. Special Ops Team
License:
Apache
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.
Latest version:
0.0.26
Required dependencies:
accelerate
|
huggingface-hub
|
numpy
|
optimum
|
protobuf
|
transformers
Optional dependencies:
black
|
compel
|
controlnet-aux
|
datasets
|
diffusers
|
gitpython
|
isort
|
libneuronxla
|
librosa
|
mediapipe
|
neuron-cc
|
neuronx-cc
|
neuronx-distributed
|
numpy
|
opencv-python-headless
|
parameterized
|
peft
|
protobuf
|
psutil
|
pytest
|
rjieba
|
ruff
|
sacremoses
|
safetensors
|
sentence-transformers
|
sentencepiece
|
soundfile
|
torch
|
torch-neuron
|
torch-neuronx
|
torchvision
|
transformers-neuronx
|
trl
|
wheel
Downloads last day:
855
Downloads last week:
10,759
Downloads last month:
44,538