Wheelodex » Projects » optimum-neuron

optimum-neuron

View on PyPI — Reverse Dependencies (1)

0.0.27 optimum_neuron-0.0.27-py3-none-any.whl

Wheel Details

Project: optimum-neuron
Version: 0.0.27
Filename: optimum_neuron-0.0.27-py3-none-any.whl
Download: [link]
Size: 318885
MD5: 53c28d0e3e84e223c5a450b800a55836
SHA256: 4f58194de16cf88e7d1197e3928999fc6883cf11fc74474b81685d79f917b80a
Uploaded: 2024-12-13 09:59:29 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt · entry_points.txt · dependency_links.txt

METADATA

Metadata-Version: 2.1
Name: optimum-neuron
Version: 0.0.27
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.
Author: HuggingFace Inc. Special Ops Team
Author-Email: hardware[at]huggingface.co
Home-Page: https://huggingface.co/hardware/aws
License: Apache
Keywords: transformers,diffusers,mixed-precision training,fine-tuning,inference,tranium,inferentia,aws
Classifier: Development Status :: 2 - Pre-Alpha
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Dist: transformers (==4.43.2)
Requires-Dist: accelerate (==0.29.2)
Requires-Dist: optimum (~=1.22.0)
Requires-Dist: huggingface-hub (>=0.20.1)
Requires-Dist: numpy (<=1.25.2,>=1.22.2)
Requires-Dist: protobuf (<4,>=3.20.3)
Requires-Dist: diffusers (<=0.30.3,>=0.28.0); extra == "diffusers"
Requires-Dist: peft; extra == "diffusers"
Requires-Dist: wheel; extra == "neuron"
Requires-Dist: torch-neuron (==1.13.1.2.9.74.0); extra == "neuron"
Requires-Dist: torch (==1.13.1.*); extra == "neuron"
Requires-Dist: neuron-cc[tensorflow] (==1.22.0.0); extra == "neuron"
Requires-Dist: protobuf; extra == "neuron"
Requires-Dist: torchvision; extra == "neuron"
Requires-Dist: numpy (==1.22.3); extra == "neuron"
Requires-Dist: wheel; extra == "neuronx"
Requires-Dist: neuronx-cc (==2.15.143.0); extra == "neuronx"
Requires-Dist: torch-neuronx (==2.1.2.2.3.2); extra == "neuronx"
Requires-Dist: transformers-neuronx (==0.12.313); extra == "neuronx"
Requires-Dist: torch (==2.1.2.*); extra == "neuronx"
Requires-Dist: torchvision (==0.16.*); extra == "neuronx"
Requires-Dist: neuronx-distributed (==0.9.0); extra == "neuronx"
Requires-Dist: libneuronxla (==2.0.5347.0); extra == "neuronx"
Requires-Dist: black; extra == "quality"
Requires-Dist: ruff; extra == "quality"
Requires-Dist: isort; extra == "quality"
Requires-Dist: sentence-transformers (>=2.2.0); extra == "sentence-transformers"
Requires-Dist: pytest (<=8.0.0); extra == "tests"
Requires-Dist: psutil; extra == "tests"
Requires-Dist: parameterized; extra == "tests"
Requires-Dist: GitPython; extra == "tests"
Requires-Dist: sentencepiece; extra == "tests"
Requires-Dist: datasets; extra == "tests"
Requires-Dist: sacremoses; extra == "tests"
Requires-Dist: diffusers (<=0.30.3,>=0.28.0); extra == "tests"
Requires-Dist: safetensors; extra == "tests"
Requires-Dist: sentence-transformers (>=2.2.0); extra == "tests"
Requires-Dist: peft; extra == "tests"
Requires-Dist: trl (==0.11.4); extra == "tests"
Requires-Dist: compel; extra == "tests"
Requires-Dist: rjieba; extra == "tests"
Requires-Dist: soundfile; extra == "tests"
Requires-Dist: librosa; extra == "tests"
Requires-Dist: opencv-python-headless; extra == "tests"
Requires-Dist: controlnet-aux; extra == "tests"
Requires-Dist: mediapipe; extra == "tests"
Provides-Extra: diffusers
Provides-Extra: neuron
Provides-Extra: neuronx
Provides-Extra: quality
Provides-Extra: sentence-transformers
Provides-Extra: tests
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 11660 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.45.1)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
optimum/commands/env.py sha256=ieqShKfmpx_6PW8K-5sUHtwjwJJOdhew8ThHVQCoHFs 3733
optimum/commands/export/neuron.py sha256=tvjQorvtml2vT1zgzGVqepXeP0wi4UIrUlcUOXloadI 6766
optimum/commands/export/neuronx.py sha256=qxLOs3_cQOtvbDkyHM6fn5kREMNF2Sdlir19zHzDLZ0 11025
optimum/commands/neuron/base.py sha256=DAknFZXjZ7i6xpc8-BDqKJqbRduzI9tdmSV9rTU-7S8 1428
optimum/commands/neuron/cache.py sha256=ohppAbjp47SV-ezpvwQ4cUxreI2WMrM_IifKLQRbi9k 9512
optimum/commands/neuron/subcommands.py sha256=FjpqNnirZpSL26CBmKZqW_6mrqK7vt_5enVW0LEwwz8 2212
optimum/commands/register/register_export.py sha256=si-xU0xUyJlvTsYl_9daUp9268CDkNeIdeMPvtwTl60 1270
optimum/commands/register/register_neuron.py sha256=IaNP3BYz2hNGD1onJUtuoslk1vl9a13LFGRINVi7QQc 755
optimum/exporters/neuron/__init__.py sha256=447lri_nGBvNzuhuDAnelW_wZ7Jym4Jffc9FiO_8jgc 2138
optimum/exporters/neuron/__main__.py sha256=VOq13EJtweLojRO2tKUs0EyKmVMHZ5oehKHnmGYwrZo 29571
optimum/exporters/neuron/base.py sha256=SRB_BDZVC_ZQCbiV8AVp4qoBRe8u40Kqxaldx92okxE 19415
optimum/exporters/neuron/config.py sha256=P7cury6ugmb_9xiCGC58w_AV9_sszRLCiAPvQrTkyWk 6022
optimum/exporters/neuron/convert.py sha256=55ATqZhsOF-kw1m_tGcxA00Mf3Dg55V9BMtnJjlDniU 34542
optimum/exporters/neuron/model_wrappers.py sha256=hd8MaF-27OkP1fqD1NL_DVOaQlBz6BD8NhlcCKtu7pQ 22553
optimum/exporters/neuron/utils.py sha256=VxAiQuee0zBVUlH4mpvJ5zaE6ul6ROIl5kfJF_JyOuM 24128
optimum/exporters/neuron/model_configs/__init__.py sha256=oj8AGMWPDI_Zsp11_yO4oyxV1m95zL-NEndkZp4s-ok 849
optimum/exporters/neuron/model_configs/decoder_configs.py sha256=3nScZfMquYH-OIizIkDoxViI4M1pxRJFLKdTp0_9it4 2191
optimum/exporters/neuron/model_configs/traced_configs.py sha256=5pzUOYQC60sqIkM0P6S1kyCC51nHnurbaEI74wWbJKY 31945
optimum/neuron/__init__.py sha256=cPgJi2pRMsPzHjZRdyz7wPO7sxhkERI8hQJLMKuLx8g 4891
optimum/neuron/hf_argparser.py sha256=uhE1LIYZ9EPDnMjp-5CnQrGbkWzTqPPe45QxiJ6Ezj4 2337
optimum/neuron/modeling.py sha256=aFTvJqcibvEuynbGnOXNYTWQD21cS7KucT8g-GSnkiM 58364
optimum/neuron/modeling_base.py sha256=b8BN2GxJvUrwaI0b4ZD1TsX0Z_S3kxvfJSTkAbFlfoo 1369
optimum/neuron/modeling_decoder.py sha256=Eo-DERIQJaRB7n6WU0QGqInPbhVNs2e9Dn_IVMT668M 19942
optimum/neuron/modeling_diffusion.py sha256=Okl_ji2dpxRtkLJhofzvFyHHUwpAbPWPNYYK7aFBZRI 70261
optimum/neuron/modeling_seq2seq.py sha256=s2a_VVahDWCW_es1Escf9zUln-rUSXHPjbAnU2W50P8 28562
optimum/neuron/modeling_traced.py sha256=z9mZ7WfTuMFG6yh5CBQmscV0wXS5Sg3jGhZHZk-DA-k 26536
optimum/neuron/trainers.py sha256=V-3YyxtzdIMhtz05mPr-6P6b5_UMuNRCsxoCVI4q35Y 104108
optimum/neuron/training_args.py sha256=7Z95cFVcKMiiuI80Q2NECA8wR5wNGbP09sDLrM9YnUQ 11250
optimum/neuron/version.py sha256=qG7Jcj0gEb-Ii_j-GGsKh4ukiGmT5n0lmmo4aoGHvVM 668
optimum/neuron/accelerate/__init__.py sha256=JwvqTR4LIX2i6yV8goGNj8k_rqj35-cRpl46lrcwn_c 809
optimum/neuron/accelerate/accelerator.py sha256=mGpaQZ3NujNPIAv1PR9fb7UDghO0aCHXsYu-9he7ifE 31153
optimum/neuron/accelerate/optimizer.py sha256=3FaxRcDgb8cAYLA_FDylsm6zetPSLNjwFyp_7bsztuQ 6270
optimum/neuron/accelerate/scheduler.py sha256=YtoYfYZLbbVhGbIY30Wy6OpTfK1Xr_JEaT9QXgiQJAI 2212
optimum/neuron/accelerate/state.py sha256=R-nFLfxtS4Y5uld_NrXhYlSR27cGuwF9-b1LzqMwqYc 10757
optimum/neuron/accelerate/utils/__init__.py sha256=zoGlRKENM4Dq3VdV1Eh1eCjKsiOW0ULMWqWGln8f0dQ 790
optimum/neuron/accelerate/utils/dataclasses.py sha256=wwW9mPW8Lo6XQZMu2cUmjIN0N5Pm2qjHCAxkk8XXhSc 3561
optimum/neuron/accelerate/utils/misc.py sha256=uGI9Q6_j4r4XB2KLQFMKB94OOHi6nhaiiDna7jUDXq0 8885
optimum/neuron/accelerate/utils/operations.py sha256=CutOduJROM3Y8PHX7b_Tja6hZn_JmbWrgjzrKxJqX1I 2542
optimum/neuron/distributed/__init__.py sha256=xFkzJY8yznzd6Xqyn8r84I2JQbVHUPuz1rGGUpzg640 933
optimum/neuron/distributed/base.py sha256=V9yeBKePs2T_14YQeqzx6BGeN9GqWcfQgg7b-Wtud6M 52081
optimum/neuron/distributed/checkpointing.py sha256=A7kuLWsJLJl1IHrYnxVPWp3Z95Kv9hsC7ff09oyv7so 11003
optimum/neuron/distributed/decoder_models.py sha256=yZjt0C_B5dSZohz10-RaThA6ITGyLTy6o2L5qdNOZYA 36377
optimum/neuron/distributed/encoder_decoder_models.py sha256=gMMGZCHwdAwiFtpyZxQ-FhUl6qSL7ToqkzbpMaDQMbI 18219
optimum/neuron/distributed/encoder_models.py sha256=4ZU8m0caMw5TPbWT1FP4RkmRf0t4mnSQ2JBnxv_icwg 11898
optimum/neuron/distributed/parallel_layers.py sha256=9GKJxP3vpbaS9aVfACevYqpu6Er_2Gs0Rt8TmpBTA24 54379
optimum/neuron/distributed/parallelizers_manager.py sha256=5TyzLlikB-mP1VZEqs7CQIAPlLp2C4oyVgIxloGrXJM 5325
optimum/neuron/distributed/utils.py sha256=czzCcUxFFHIOK33Yd5HZE_06bbeiVabOMCTI7EuX7uA 76509
optimum/neuron/generation/__init__.py sha256=QEDIn7WV8p3d25XlOcEe9poqe-mt69PDZ1yV_hiSu0k 786
optimum/neuron/generation/logits_process.py sha256=m2a8cFq589UumoC3ercmZ0kGjIaYHO4dmpQwOaABw5A 3862
optimum/neuron/generation/token_selector.py sha256=JZBpvrhvIZTTJiwbbnEtL9RahhKo8bfmAU9qtT-HnIg 8798
optimum/neuron/generation/utils.py sha256=WKxUApCqlMjYIKOwxkLJLqVAwUSHHf2UN7mfE2LtlKQ 83583
optimum/neuron/models/__init__.py sha256=TGVQwFyxkb7pGSbC0fq2wFHHY-agemZr4UFi6bgpK0U 621
optimum/neuron/models/qwen2/__init__.py sha256=TGVQwFyxkb7pGSbC0fq2wFHHY-agemZr4UFi6bgpK0U 621
optimum/neuron/models/qwen2/config.py sha256=2QePvx7GK_kylDBpMAEoMR_-SXb59aTTu-oEc3d9QWk 1087
optimum/neuron/models/qwen2/model.py sha256=QwvESiG0QwGIMYMvXMnOp07CjhFLMCmWSEXQnxsAtkk 15193
optimum/neuron/models/qwen2/modules.py sha256=twpw2hytmez2AaPpGGxWZCubAUDUgSEy0biF7aBMxUc 3467
optimum/neuron/pipelines/__init__.py sha256=TqfFMj-WOJQCEepPesEEJFKKasPhF-ivY8VVDw__B6o 1371
optimum/neuron/pipelines/diffusers/__init__.py sha256=9PG4dT13AaYLM612COpw0p6wD3Zglmq_tuxugDHv9LM 850
optimum/neuron/pipelines/diffusers/pipeline_controlnet.py sha256=D2smmWC-XJrppUboaa4lDW3_wjzChcMKuKXrMWgy4wI 25082
optimum/neuron/pipelines/diffusers/pipeline_controlnet_sd_xl.py sha256=31PmICVptkwe10FxmevxJL6f5zE_neiCLmELFt8miag 34436
optimum/neuron/pipelines/diffusers/pipeline_utils.py sha256=9T8klNp1xs76_yNzMzgD91fM-MW73k8OQoEanNL990g 3246
optimum/neuron/pipelines/transformers/__init__.py sha256=RXpHlx0tILCeYroRAx0jlNBQztPJnTxbF4zwn4zRbVI 658
optimum/neuron/pipelines/transformers/base.py sha256=aW5cN3WFnusCyk37adr97xYM8V1eWM3UtkIBs8sXMfY 12348
optimum/neuron/pipelines/transformers/sentence_transformers.py sha256=eXKHzM_l8vPaE_W4X1PqofVAm5c2abTxvmjJ-WWJC7E 3546
optimum/neuron/utils/__init__.py sha256=bt-eUuKVKL_IOu0qYYyd-f7_5w1rxU_2L-8bW9V5eaM 4808
optimum/neuron/utils/argument_utils.py sha256=4XrFVLQra9KJfhEEKtqzYA7Imf1R4m7Q8kwXTTs8gfc 7875
optimum/neuron/utils/cache_utils.py sha256=cGezUOMqjdtjEQcBIN5FAeUBkL_z_8gAhb9_lL4UDL8 10739
optimum/neuron/utils/constant.py sha256=jE8HU1nszV5xYEfehXWQ_qBhpIR2v4o5YNDuBjIoNfo 1132
optimum/neuron/utils/deprecate_utils.py sha256=ML7shIzgsPRTAJGzorIZ8eOAAJQ1HevSZeuh8cGp_RQ 2720
optimum/neuron/utils/hub_cache_utils.py sha256=eRbijmRtV0OHMOevdR3T-vnQ5wLhd_NUGlhh7K2vuNA 23227
optimum/neuron/utils/import_utils.py sha256=tjKQR-Ba6wyIdfrVqNC4Vz4wLmIWoaSIHHZrB7HR_cg 2606
optimum/neuron/utils/input_generators.py sha256=Hn6ZK6v7qQePw2jIQLzHKvBAAhxf6I98Ss7u17KaT0c 7521
optimum/neuron/utils/misc.py sha256=a8YBMUZuB2m3q69Ustb0riFOC2PNvNPLuhu9VmEPGs0 30587
optimum/neuron/utils/model_utils.py sha256=GuFGepo7p028zTHX9AT49JP9VOJ5LgmSiU3d952oMbo 3215
optimum/neuron/utils/neuron_cc_wrapper sha256=kHMw49tfWp0CtEzGEAyQTxrejAatk4ejALBkw3-8p9s 250
optimum/neuron/utils/neuron_parallel_compile.py sha256=uAmT7JgLeQF2E-hk1xnzCXxBsQ5bwLowNHp4cIOVFf8 1290
optimum/neuron/utils/optimization_utils.py sha256=JpPTBlYW8KexJOn5lFdqDrChfFZQXtbXkipCOiWBidI 2281
optimum/neuron/utils/optimum_neuron_cc_wrapper.py sha256=QKto-cp9-2R3xoWIuvbVgpd1_P0YpLcy0Yed_eNXVKw 1027
optimum/neuron/utils/patching.py sha256=jaO4otCwrMqiy_IsaPFSzTZXDWfbFzzBlWplwiWXNSk 9255
optimum/neuron/utils/peft_utils.py sha256=0HummoLJIaZRBFpd6teUWY23xYp-On0cpeNel2TkCdU 13086
optimum/neuron/utils/require_utils.py sha256=apdCJT6YHIoz_aQ1RHAcT2EplQdSegJT8T_satc1x7Q 2322
optimum/neuron/utils/runner.py sha256=K6XDlJS2BYImTTzj_UmpqdF2MtqvfFqgM5kD8ZlIkik 25666
optimum/neuron/utils/testing_utils.py sha256=BZTw0yig6fc8cWXWBm-cwFLtUELOAEIYUP1wCb2HVk4 1644
optimum/neuron/utils/torch_xla_and_neuronx_initialization.py sha256=_-WBgb7BWTmDRzd2aBhn2HaJZevM3xOMgo3HBLPTra4 4043
optimum/neuron/utils/training_utils.py sha256=9787Fhby_XjfTJxwciCt8jyEfFbyiiwXxuxKGQNIbUM 11307
optimum/neuron/utils/trl_utils.py sha256=DFEIJY0usOCUSkK23_ji5uZXVOIduZ91DFixXmFINRk 1532
optimum/neuron/utils/version_utils.py sha256=4CoANMOaVOxim4K_SeNSZvmx3GhcnJXQTqGPoLj--rY 4377
optimum_neuron-0.0.27.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
optimum_neuron-0.0.27.dist-info/METADATA sha256=MFjxDLLmKIiBxVx6aJpq7IfqVYB9BQQYiEs6QTrSzCQ 15215
optimum_neuron-0.0.27.dist-info/WHEEL sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A 92
optimum_neuron-0.0.27.dist-info/dependency_links.txt sha256=4v7Jf0ylcHEOE-oPSq6kbH5Pk7kl196QnJjeV89UF4c 39
optimum_neuron-0.0.27.dist-info/entry_points.txt sha256=sgPkvCyde_KUZXpAF6hN7pzOBej-8BevY3PdTmzeYoI 142
optimum_neuron-0.0.27.dist-info/top_level.txt sha256=rc1rV-uPZnV1Ek7hCwh56pvMCKcip65JbHRRWs8Yqu8 8
optimum_neuron-0.0.27.dist-info/RECORD — —

top_level.txt

optimum

entry_points.txt

[console_scripts]
neuron_parallel_compile = optimum.neuron.utils.neuron_parallel_compile:main
optimum-cli = optimum.commands.optimum_cli:main
https://pip.repos.neuron.amazonaws.com