Wheelodex » Projects » optimum-neuron

optimum-neuron

View on PyPI — Reverse Dependencies (1)

0.0.21 optimum_neuron-0.0.21-py3-none-any.whl

Wheel Details

Project: optimum-neuron
Version: 0.0.21
Filename: optimum_neuron-0.0.21-py3-none-any.whl
Download: [link]
Size: 310568
MD5: d8d42676ebfd924b0182bd3f13af6f5d
SHA256: bc892078a4ac9b4cf9c0a20cfc24930c72360b6cb5567438d39ad25fec1e5ad7
Uploaded: 2024-04-08 22:45:44 +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.21
Summary: Optimum Neuron is the interface between the Hugging Face Transformers and Diffusers libraries and AWS Tranium 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.36.2)
Requires-Dist: accelerate (==0.23.0)
Requires-Dist: optimum (~=1.18.0)
Requires-Dist: huggingface-hub (>=0.20.1)
Requires-Dist: numpy (<=1.25.2,>=1.22.2)
Requires-Dist: protobuf (<4)
Requires-Dist: diffusers (~=0.26.1); 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: wheel; extra == "neuronx"
Requires-Dist: neuronx-cc (==2.13.66.0); extra == "neuronx"
Requires-Dist: torch-neuronx (==1.13.1.1.14.0); extra == "neuronx"
Requires-Dist: transformers-neuronx (==0.10.0.21); extra == "neuronx"
Requires-Dist: torch (==1.13.1.*); extra == "neuronx"
Requires-Dist: torchvision (==0.14.*); extra == "neuronx"
Requires-Dist: neuronx-distributed (==0.7.0); extra == "neuronx"
Requires-Dist: black; extra == "quality"
Requires-Dist: ruff; extra == "quality"
Requires-Dist: isort; extra == "quality"
Requires-Dist: hf-doc-builder; 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.26.1); extra == "tests"
Requires-Dist: safetensors; extra == "tests"
Requires-Dist: sentence-transformers (>=2.2.0); extra == "tests"
Requires-Dist: peft; 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: 11610 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.43.0)
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=ls-0YYxZ9VJQXHrfqVMvnV8hFg-mT1upnIfZ7Pj8uxU 7045
optimum/commands/export/neuronx.py sha256=IUXBmLDcJ3dREeskN-PrSRkBk5GbOiyWu0elu5vJrrk 10443
optimum/commands/neuron/base.py sha256=DAknFZXjZ7i6xpc8-BDqKJqbRduzI9tdmSV9rTU-7S8 1428
optimum/commands/neuron/cache.py sha256=-3R1408ya71g65Vr5hwxAiarsyALt6CxjTGBvuv1Ts0 9350
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=-tVawOlOLgFCg2fymo32O9hBpnIT5KTgUEN_oon1x78 26954
optimum/exporters/neuron/base.py sha256=Jwws4YdCGs7qi8trGSj2TLoRjK59feSRCqLWerXxQ-Q 16943
optimum/exporters/neuron/config.py sha256=spCfWHGoC5N_fG6kGCRhcxhXgnsAUEGaNedgVHj23zQ 5749
optimum/exporters/neuron/convert.py sha256=pPpQBs-9y0nAW_ZZfS14zCUr1Lbit0AYY-lOtBttvu0 32835
optimum/exporters/neuron/model_configs.py sha256=YaeWKBXqYUNLrX-QmjulmRPl3g0cb2YuTBhZxyPcANA 19542
optimum/exporters/neuron/model_wrappers.py sha256=JQapl5tVwNeGfIyLEDx-M61Y2cgriSTyxVvlo80dWTo 15517
optimum/exporters/neuron/utils.py sha256=CpEsGFg0o6z_LzN7T5dNK5acVzy7EK7vKsfCJW_MNWM 20653
optimum/neuron/__init__.py sha256=UwuiR53Hufj1JG9V0if9aKjSHCVtq8lkgnzjASGjE8E 3626
optimum/neuron/hf_argparser.py sha256=uhE1LIYZ9EPDnMjp-5CnQrGbkWzTqPPe45QxiJ6Ezj4 2337
optimum/neuron/modeling.py sha256=XySpNYy1FM1gQVnU0YLlmCEcxBMQWDKu9WtfLEh-U_M 37969
optimum/neuron/modeling_base.py sha256=SvWwSvhYtrz70H4_kOcNkkBFvIsECagk5nj8d8rxluY 25671
optimum/neuron/modeling_decoder.py sha256=M2noZwQrQBBqDvsd65ogp2aC64_0ky6rLkIVfLU1G0s 19813
optimum/neuron/modeling_diffusion.py sha256=Mci-PKjqd_9FcN0VUzBw4c-d7HQbKNXzLbeY-ab7tqw 47872
optimum/neuron/modeling_seq2seq.py sha256=un1vQb2uNCRauEQHyLa3SBoxrDtvKlChrLUMMaBtFp0 23525
optimum/neuron/trainers.py sha256=xGQb_JxljV9Fpyi5T2J8VccU1oWQFO2gwMNDcViJ5cQ 66785
optimum/neuron/training_args.py sha256=0H_ehZxDcZKo9O8kFMTN77wgC0gitN9Oolb8ZHFMXJY 13520
optimum/neuron/version.py sha256=qr8LV95hz54avceISzBNb3748i755QhdMTBI_E-tXX4 668
optimum/neuron/accelerate/__init__.py sha256=JwvqTR4LIX2i6yV8goGNj8k_rqj35-cRpl46lrcwn_c 809
optimum/neuron/accelerate/accelerator.py sha256=0EdVaySBfedGnTCPPmOdz0c5SmxjbssBB3uYQvs19aU 34174
optimum/neuron/accelerate/optimizer.py sha256=ERXDDq9YyIJzVhwDR11F9fWO_hy8HRSX2ruWCrhHKAg 6407
optimum/neuron/accelerate/scheduler.py sha256=YtoYfYZLbbVhGbIY30Wy6OpTfK1Xr_JEaT9QXgiQJAI 2212
optimum/neuron/accelerate/state.py sha256=F__tHY3WSSNGfj9Wf2gm7CwPoXynxAjc0b7yiJaPxlc 18128
optimum/neuron/accelerate/utils/__init__.py sha256=zkmrNWm2TMtMGNc6ondw9JWx7smTL9Zv0l9Od_ZOaKo 868
optimum/neuron/accelerate/utils/dataclasses.py sha256=t8645UxSGOfXN68ewXjyol4JWjx0j42b0TZZT1GK928 9133
optimum/neuron/accelerate/utils/misc.py sha256=rSCr36j5aSG2ljSf1UM7mqR07Y1dTt0SWWmfptPPU-A 3861
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=6h5MZFVUBOREXvQVgKfpe3r9cmMLrb_YO1aRdKTF44w 51128
optimum/neuron/distributed/checkpointing.py sha256=oDPB0Sk_XkG4s9epFEETHIl0wDwedVWXSyZ5gseiKpY 8945
optimum/neuron/distributed/decoder_models.py sha256=HiXbCi_aC-MdjX2lYspgoZwyxD1yicLPV99mu2f9e0U 38350
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=xYNsWEQu1WOWUn1nVn9pt90nNGT3kuy2MOUANH_ui08 53429
optimum/neuron/distributed/parallelizers_manager.py sha256=uv3EVeJVoQP0kw6RW4E7WwQuvNfOPr4Pd9wsTzSg6NA 4388
optimum/neuron/distributed/utils.py sha256=LnbI9L6ZvzSEymuUzxuap-Q35CRCSGTvXLPAO0UO9ZE 65859
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=RX4Rs2DMh6777x9eqV9PppLBCKO5f7Rv7o2u_w9dy-k 8326
optimum/neuron/generation/utils.py sha256=DyEyXeJLA-gVXlFf8IY14L3YZS7u72wgzWg1c936zLU 83934
optimum/neuron/pipelines/__init__.py sha256=CDqbkWnj2Pollq_vK_dE9NjwzBw5tA2C0AaGjXi7_hU 1772
optimum/neuron/pipelines/diffusers/__init__.py sha256=NcFDudIjKrXfs6ugF4HHZhbZQ4dCSOmM2mvmChoNOIo 1228
optimum/neuron/pipelines/diffusers/pipeline_latent_consistency_text2img.py sha256=cpgKZCLOya86CM3R0pBAvw89bcFmetJSwssl3Kaw1q8 15197
optimum/neuron/pipelines/diffusers/pipeline_stable_diffusion.py sha256=O32ZgUisEG_Bz_AgbQES9H2SJfdbUddpDFok0LXGQzI 14205
optimum/neuron/pipelines/diffusers/pipeline_stable_diffusion_img2img.py sha256=_z1sABNkplf5Nk6NgMQneyFSNc6t9AhexqSMPNi5nwg 17259
optimum/neuron/pipelines/diffusers/pipeline_stable_diffusion_inpaint.py sha256=Ubo9M23Ip5hx4O3VQmJF54oZP6Z_xaNT8RWSPiSMR1I 22196
optimum/neuron/pipelines/diffusers/pipeline_stable_diffusion_xl.py sha256=ry335mcELb05xfjakb8LZZzUlnnWroMTnuPaVUUtaP8 23491
optimum/neuron/pipelines/diffusers/pipeline_stable_diffusion_xl_img2img.py sha256=TWYuVamiTklWagQW_6CEnVcWR8gxlVycmrt4n_4zLb0 29826
optimum/neuron/pipelines/diffusers/pipeline_stable_diffusion_xl_inpaint.py sha256=UPHXa_QXGjcl_H9Y5ySE3PCQpQlGGsQk-VatTReBBOA 39059
optimum/neuron/pipelines/diffusers/pipeline_utils.py sha256=hjieSfHCRhvp_L2TrvhuJClEcM_vHTg3Do_ovlYursY 22025
optimum/neuron/pipelines/transformers/__init__.py sha256=RXpHlx0tILCeYroRAx0jlNBQztPJnTxbF4zwn4zRbVI 658
optimum/neuron/pipelines/transformers/base.py sha256=POWZ0CGTU19gvlfXsLIAsp74JS_hvhlkh01NJDOzKGA 9860
optimum/neuron/utils/__init__.py sha256=Ipek1pag9889Qm6nCq02sXvokGp1ntdypQOQF7PHksU 1904
optimum/neuron/utils/argument_utils.py sha256=E2fwK9jNKiVuUdnZ5TguPIPMnEyZwvFpFJBcIx3Mez0 7767
optimum/neuron/utils/cache_utils.py sha256=oyXJLx5_Ap3Tk6g5mveBJtdu1mAs1n4VNKCV-NKaVOk 10646
optimum/neuron/utils/constant.py sha256=tlqsh6sYPBKZTv1qss81FFPGLYe9BpunFm6nimtbl_c 1036
optimum/neuron/utils/deprecate_utils.py sha256=ML7shIzgsPRTAJGzorIZ8eOAAJQ1HevSZeuh8cGp_RQ 2720
optimum/neuron/utils/hub_neuronx_cache.py sha256=pJ0et57tK-LmFDZBAIAcM1YkFi4K7NQEE6iJIfwRq5k 23166
optimum/neuron/utils/import_utils.py sha256=TDSsxoQKWEg57E9aVmsH0lcoJIN_k4C5d06a1OeTH-c 2111
optimum/neuron/utils/input_generators.py sha256=C05fmuIlGsFFvWZKHl2sI856PWg6JVFadTZGgA31M7k 1529
optimum/neuron/utils/misc.py sha256=jBuz-DUBXdwXQ8pN2pdJkCEvuqVwXOXgPW4b0KMNh7w 30575
optimum/neuron/utils/neuron_cc_wrapper sha256=kHMw49tfWp0CtEzGEAyQTxrejAatk4ejALBkw3-8p9s 250
optimum/neuron/utils/optimization_utils.py sha256=JpPTBlYW8KexJOn5lFdqDrChfFZQXtbXkipCOiWBidI 2281
optimum/neuron/utils/optimum_neuron_cc_wrapper.py sha256=RKZtpo3nQBr5ARmMm32yOrlEZBa3xWE1KKWShbnuyk4 1029
optimum/neuron/utils/patching.py sha256=xZpLFzhq9lfaXe8hl-lEcWaG36ZUdEvYb5w5ulj-lEM 8412
optimum/neuron/utils/require_utils.py sha256=BCrro0ZUhh_UB5rLqC-3o-FjSGzorBZJoKAbTPM7JKM 2232
optimum/neuron/utils/runner.py sha256=DehQyKK57-hOo1ZAGL1BlH0BGfAQ7eFW3XMCwolAxzA 24186
optimum/neuron/utils/testing_utils.py sha256=BZTw0yig6fc8cWXWBm-cwFLtUELOAEIYUP1wCb2HVk4 1644
optimum/neuron/utils/torch_xla_and_neuronx_initialization.py sha256=4MljtbRIM_F-1sg6WNU1_JozCXMZEnJut1vGBgpyrY8 3964
optimum/neuron/utils/training_utils.py sha256=7CggZM3TKHQqXzkqdBcHjU5ei23YdGkzIwoyNa7qAME 15025
optimum/neuron/utils/version_utils.py sha256=4CoANMOaVOxim4K_SeNSZvmx3GhcnJXQTqGPoLj--rY 4377
optimum_neuron-0.0.21.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
optimum_neuron-0.0.21.dist-info/METADATA sha256=Bu036ufirtQv4e2gUG0_ROWHHt8H9Qe2E_Ajl8Qo5t4 14777
optimum_neuron-0.0.21.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
optimum_neuron-0.0.21.dist-info/dependency_links.txt sha256=4v7Jf0ylcHEOE-oPSq6kbH5Pk7kl196QnJjeV89UF4c 39
optimum_neuron-0.0.21.dist-info/entry_points.txt sha256=1__lv2jpNhLb5ebCqqbs8FFPhOfi9_3nH1dP1lvvZ9s 66
optimum_neuron-0.0.21.dist-info/top_level.txt sha256=rc1rV-uPZnV1Ek7hCwh56pvMCKcip65JbHRRWs8Yqu8 8
optimum_neuron-0.0.21.dist-info/RECORD — —

top_level.txt

optimum

entry_points.txt

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