openteamsinc
pypi/optimum-neuron
Risk Profile
Package
Source
Risk Profile
Maturity:
Unknown
Repo not found
Health:
Unknown
Repo not found
Legal:
Unknown
Repo not found
Security:
Unknown
Repo not found
Package
pypi
optimum-neuron
Version
0.3.0
Last Release Date
about 2 months ago
License
Apache
Dependencies
transformers
~=4.51.0
accelerate
==1.8.1
optimum
~=1.24.0
huggingface-hub
>=0.29.0
numpy
<=1.25.2, >=1.22.2
protobuf
<4, >=3.20.3
diffusers:
diffusers
<=0.34.0, >=0.31.0
peft
==0.16.0
neuron:
wheel
torch-neuron
==1.13.1.2.9.74.0
torch
==1.13.1.*
neuron-cc
[tensorflow]
==1.22.0.0
protobuf
torchvision
numpy
==1.22.3
neuronx:
wheel
neuronx-cc
==2.19.8089.0
torch-neuronx
==2.7.0.2.8.6734+ac864f72
torch
==2.7.0.*
torchvision
==0.22.*
neuronx-distributed
==0.13.14393
libneuronxla
==2.2.4410.0
quality:
ruff
isort
sentence-transformers:
sentence-transformers
>=2.2.0
tests:
pytest
<=8.0.0
pytest-rerunfailures
cloudpickle
pytest-timeout
pytest-order
psutil
parameterized
GitPython
sentencepiece
datasets
sacremoses
diffusers
<=0.34.0, >=0.31.0
safetensors
sentence-transformers
>=2.2.0
peft
==0.16.0
trl
==0.11.4
compel
rjieba
soundfile
librosa
opencv-python-headless
controlnet-aux
mediapipe
timm
>=1.0.0
hf-transfer
torchcodec
training:
trl
==0.11.4
peft
==0.16.0
evaluate
==0.4.3
neuronx-distributed
==0.13.14393
vllm:
vllm
==0.9.2
Source
Location
huggingface.co/docs/optimum-neuron/index
Last Source Update
n/a
Licenses
Distribution Destinations
