openteamsinc
pypi/optimum-neuron
Risk Profile
Package
Source
Risk Profile
Maturity:
Unknown
Source code is in a private repository
Health:
Unknown
Source code is in a private repository
Legal:
Unknown
Source code is in a private repository
Security:
Unknown
Source code is in a private repository
Package
pypi
optimum-neuron
Version
0.2.2
Last Release Date
16 days ago
License
Apache
Dependencies
transformers
~=4.51.0
accelerate
==0.29.2
optimum
~=1.23.3
huggingface-hub
>=0.29.0
numpy
<=1.25.2, >=1.22.2
protobuf
<4, >=3.20.3
diffusers:
diffusers
<=0.30.3, >=0.28.0
peft
==0.14.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.17.194.0
torch-neuronx
==2.5.1.2.6.0
torch
==2.5.1.*
torchvision
==0.20.*
neuronx-distributed
==0.11.0
libneuronxla
==2.2.1630.0
quality:
ruff
isort
sentence-transformers:
sentence-transformers
>=2.2.0
tests:
pytest
<=8.0.0
pytest-rerunfailures
pytest-timeout
psutil
parameterized
GitPython
sentencepiece
datasets
sacremoses
diffusers
<=0.30.3, >=0.29.0
safetensors
sentence-transformers
>=2.2.0
peft
==0.14.0
trl
==0.11.4
compel
rjieba
soundfile
librosa
opencv-python-headless
controlnet-aux
mediapipe
timm
>=1.0.0
hf-transfer
training:
trl
==0.11.4
peft
==0.14.0
evaluate
==0.4.3
neuronx-distributed
==0.10.1
Source
Location
huggingface.co/hardware/aws
Last Source Update
n/a
Licenses
Distribution Destinations
