PyPI page
Home page
Author:
None
License:
Apache-2.0
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.
Latest version:
0.4.5
Required dependencies:
huggingface_hub
|
optimum
|
transformers
Optional dependencies:
accelerate
|
cloudpickle
|
compel
|
controlnet-aux
|
datasets
|
diffusers
|
docker
|
evaluate
|
gitpython
|
isort
|
libneuronxla
|
librosa
|
neuronx-cc
|
neuronx_distributed
|
numpy
|
openai
|
parameterized
|
peft
|
pre-commit
|
protobuf
|
psutil
|
pytest
|
pytest-asyncio
|
pytest-forked
|
pytest-rerunfailures
|
pytest-timeout
|
rjieba
|
ruff
|
sacremoses
|
safetensors
|
sentence-transformers
|
sentencepiece
|
soundfile
|
torch
|
torch-neuronx
|
torchcodec
|
torchvision
|
trl
|
vllm
|
wheel
Downloads last day:
185
Downloads last week:
1,372
Downloads last month:
3,952