Wheelodex » Projects » ai-edge-torch-nightly

ai-edge-torch-nightly

View on PyPI — Reverse Dependencies (0)

0.3.0.dev20241225 ai_edge_torch_nightly-0.3.0.dev20241225-py3-none-any.whl
0.3.0.dev20241224 ai_edge_torch_nightly-0.3.0.dev20241224-py3-none-any.whl

Wheel Details

Project: ai-edge-torch-nightly
Version: 0.3.0.dev20241224
Filename: ai_edge_torch_nightly-0.3.0.dev20241224-py3-none-any.whl
Download: [link]
Size: 335358
MD5: a305fd9a2a70db9489b7d72027f80df4
SHA256: 05cfb38a1d9a12133caaa843ba4db217a9b899e7f9c7e89edc0b1f8e95355c62
Uploaded: 2024-12-24 10:14:31 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: ai-edge-torch-nightly
Version: 0.3.0.dev20241224
Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
Home-Page: https://github.com/google-ai-edge/ai-edge-torch
Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.10
Requires-Dist: numpy
Requires-Dist: scipy
Requires-Dist: safetensors
Requires-Dist: tabulate
Requires-Dist: torch (>=2.4.0)
Requires-Dist: tf-nightly (>=2.19.0.dev20241201)
Requires-Dist: ai-edge-litert-nightly
Requires-Dist: ai-edge-quantizer-nightly
Requires-Dist: jax
Requires-Dist: torch-xla2[odml] (>=0.0.1.dev20241201)
Requires-Dist: torch-xla (>=2.4.0); extra == "torch-xla"
Provides-Extra: torch-xla
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 420 characters]

WHEEL

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

RECORD

Path Digest Size
ai_edge_torch/__init__.py sha256=rq9ZtMJLG8yYNC4tNE4rpl94UAUClZW7f4GAr6HBVDQ 1208
ai_edge_torch/_config.py sha256=QIrerb6uHMahRvMilmhodJ_6jfiRps3qgLOBeidPnS4 1614
ai_edge_torch/conftest.py sha256=r0GTrhMRhlmOGrrkvumHN8hkmyug6WvF60vWq8wRIBI 758
ai_edge_torch/fx_pass_base.py sha256=518ziQ0TUxqum2qZXqlD8qr65pHPh8ZNLnwFC6zvK3k 4253
ai_edge_torch/model.py sha256=N-pNpTxzhaFGhWhnSGd70lBzb9VlEhTOq5mddU7bvvI 5542
ai_edge_torch/version.py sha256=TkfJYt2lJC8A_AcieO1xVmMQ2xdnoTOwF8CZ5dZeaqc 706
ai_edge_torch/_convert/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/_convert/conversion.py sha256=SzbR16V2JEfkCjjPwRVAFUbFnzu-_1iHPKgGT9Yz7gQ 5678
ai_edge_torch/_convert/conversion_utils.py sha256=Sr8qXVcTwc-ZnZmK7yxVrIOOp1S_vNrwzC0zUvLTI2o 2160
ai_edge_torch/_convert/converter.py sha256=075F8LRewk_033Ebsnft7FJr3KgtIbtZ_-8udIPy6ho 9980
ai_edge_torch/_convert/signature.py sha256=rGpBNss3Y9FCCCcdBwDo16KqavJi8N5P0M_6Wnmoz6Y 2320
ai_edge_torch/_convert/to_channel_last_io.py sha256=_31phf7TYgZY2ftpNbrdlB1RhDium1lz_BXEQ6IsMFc 2893
ai_edge_torch/_convert/fx_passes/__init__.py sha256=NVe-eGcm7j8jZpP2pcMhC8j5dVjgR1pPzyXhHdvKH4E 1267
ai_edge_torch/_convert/fx_passes/build_aten_composite_pass.py sha256=doaww8KqrgRTD5LotBVAIRFsEqzPn9R5lcGehBJOczA 9098
ai_edge_torch/_convert/fx_passes/build_interpolate_composite_pass.py sha256=qb4JBDi4Xca14JJUIcaaZQIJiyqKyHJF49jsRCIFCVA 4335
ai_edge_torch/_convert/fx_passes/inject_mlir_debuginfo_pass.py sha256=IlZuK42kfVcRqAWZp4j2k_81T2uWo9T2558U_GPJAlU 2327
ai_edge_torch/_convert/fx_passes/remove_non_user_outputs_pass.py sha256=f1IUVWyhioOClsMiZzLyynoW2R17U83vA-7Q-3pGPM4 2126
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/__init__.py sha256=lxnoH-WGLeiQIF8XjMGodjiZEFTxucl7g05N7MR9OPk 796
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_check.py sha256=hDsl9AHzmyuSWsdHOSO114l4nBUgUdAOUWafMTipMgA 7629
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_mark.py sha256=4RyGUwR22bZqkn_TnptenFJodc_Q43f4_SBG7gmTbos 1621
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_rewrite.py sha256=NW37V6QYdPOZOVhqLcmssVk-VAeO4ECk_CrbEBh4B0E 12740
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/op_func_registry.py sha256=bsYnudRlXp1PJlu4GF25KSogSkBGQPSaecBrUTONKaw 1031
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/pass_body.py sha256=t94Am3iPbYQekg-rrtc-jS_aDWtEgAAj7pAKHrG0-9U 10563
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/utils.py sha256=YLMttMg5PdvXTtQ8lxpKb434UGVvYVALV1-xeuH4UGc 2131
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/__init__.py sha256=D8VX8SbCzfoyvPgMFHK7yxD7R-bzLxp2gfdKxgrWekA 742
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/greedy.py sha256=L_x8BrF7UDah-SYl-pG11I6CIckdU9kBTUHcmwW4cts 2420
ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/min_cut.py sha256=mzfL9cf0qBnpmxM_OlMQFvQsEZV2B_Mia9yEJV4J7rI 7135
ai_edge_torch/_convert/test/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/_convert/test/test_convert.py sha256=gK9QJuLbpjXt0l6tVnzl9Miq6GLkJR-hB67i3VE13Og 17224
ai_edge_torch/_convert/test/test_convert_composites.py sha256=BCIODgxMI_3MxMLfNWYMGjcz-al-J3z5eDHCiZJXNwY 7992
ai_edge_torch/_convert/test/test_convert_multisig.py sha256=6_C2R9--KyNR7_oezZIAfyTSR97tOeEWy4XGcbSxBDE 5778
ai_edge_torch/_convert/test/test_to_channel_last_io.py sha256=1o-gUiwzIuO67FNAJ8DeyKv8fVUeZVNNNwofNVDjYeU 3024
ai_edge_torch/debug/__init__.py sha256=N05Mmvi41KgSuK0JhuMejERESgP8QekiGdp9_PEyuKU 742
ai_edge_torch/debug/culprit.py sha256=7UYVpVWpiCXbMAyThVtHt_kc_poT7sCTh5UUPvcycgk 14832
ai_edge_torch/debug/utils.py sha256=vOAL4t6Lj47uhKapfEsc_WHmvwew3eKO9hSJyzvPXnU 1625
ai_edge_torch/debug/test/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/debug/test/test_culprit.py sha256=fRN-8jJicawJ2mhPRQNAQUZ8AdGg-s0tYMXyhnLAlWw 3875
ai_edge_torch/debug/test/test_search_model.py sha256=-RuU0QsjqkfzZF2IbeA55MoeVOawhbgiSEu96PmioPE 1668
ai_edge_torch/experimental/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/amd_llama_135m/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/amd_llama_135m/amd_llama_135m.py sha256=urNif89PyCXbdXT5spOeDvdM5luJ-a5HaXHM86v4JnU 2766
ai_edge_torch/generative/examples/amd_llama_135m/convert_to_tflite.py sha256=Oqlg5ZoUuG2aU3067QaPpmEXWOdB8GEq7u_NWoBpoB4 2337
ai_edge_torch/generative/examples/amd_llama_135m/verify.py sha256=-9Nb9D818YSJR3olVtBwoLNeMMD5qE58YBnsA67hlHg 2421
ai_edge_torch/generative/examples/gemma/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/gemma/convert_gemma1_to_tflite.py sha256=GhwtQZ1xuMyKJl8qdxU6uKavQnlm5US9xhKJvdmgACc 2309
ai_edge_torch/generative/examples/gemma/convert_gemma2_to_tflite.py sha256=hsy4Gd7Inchi0p_Cc5yecH6vr9A7X4MvmQNfTt8N2sQ 2311
ai_edge_torch/generative/examples/gemma/gemma1.py sha256=w8oWYibZzvEvCDyp39EYyAWmjgJljhzdYPyFCfAWxZA 3497
ai_edge_torch/generative/examples/gemma/gemma2.py sha256=VTM2nO3TqK2d1DyEb2MiHc-Tyw2lMcUXyOhvg0H5ENY 10147
ai_edge_torch/generative/examples/gemma/verify_gemma1.py sha256=ip-Gmk4CI5f0GWSdAIdrectxQWJ0t328KCsA4nfHuGg 1736
ai_edge_torch/generative/examples/gemma/verify_gemma2.py sha256=IoBhEMwH07-tFm5-U6F2hpCsI8xynglhq1x9tIOdaPQ 1322
ai_edge_torch/generative/examples/gemma/verify_util.py sha256=tR8RflXocDZqvuStyw9aFlzuiTllEC8rNnjrxms6_Is 5727
ai_edge_torch/generative/examples/llama/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/llama/convert_to_tflite.py sha256=ck7tXN0U25wAbbRjDcf-aqiS2YhismkmoZIsMpjIsjc 2536
ai_edge_torch/generative/examples/llama/llama.py sha256=BMjpdw6oOXmtqXCAfW9o7Iewaj-Hxd57xVrvSLBuHTk 6656
ai_edge_torch/generative/examples/llama/verify.py sha256=X7oKQi85M789ugBrOlMvzk8eSRR3Kf1Mprfl-U-WIpo 2842
ai_edge_torch/generative/examples/moonshine/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/moonshine/convert_moonshine_to_tflite.py sha256=7m3rYRzThRDYb-7pGnpLr3ACi4PWX07Mg20Q98ArPc4 1714
ai_edge_torch/generative/examples/moonshine/moonshine.py sha256=nZ2b8u4TmsB5sgdClgAuH8E78bcTv9RCnF9666HqP2M 3394
ai_edge_torch/generative/examples/openelm/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/openelm/convert_to_tflite.py sha256=nji1oDgf6xImvGh95--8cNl3QPs-Xml2XBgNJB_c2hY 2323
ai_edge_torch/generative/examples/openelm/openelm.py sha256=sIJ8Ie1oxFrJM-1jvv2ukiJbQOTIUGuMEZvmwZbt3n0 4556
ai_edge_torch/generative/examples/openelm/verify.py sha256=VkigoqhAr8ew95neb3TifYv-SLOSheaWKv2AH0iKDrc 2441
ai_edge_torch/generative/examples/paligemma/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/paligemma/convert_to_tflite.py sha256=LFCcnkmOksySDa_5bLBzoGMijYdFVjXIMidUlyzAbNk 2996
ai_edge_torch/generative/examples/paligemma/decoder.py sha256=amN96oBMTPolOFvGa47vG92AZ-BNLm8j0bBYd-IrMvI 5407
ai_edge_torch/generative/examples/paligemma/decoder2.py sha256=0V_CX0Pn5Fj_-koOGjc_Av2KMSAaVjAlD-G8P6FBGyY 6385
ai_edge_torch/generative/examples/paligemma/image_encoder.py sha256=yKPWG8aBp-GuzeyQntlzwTTcGBBjvUywVGRjnlNprmo 5574
ai_edge_torch/generative/examples/paligemma/paligemma.py sha256=FwGlFHl9zktGDxnoOpEtbS6NYN5RyzcOXH7lvNUCwEU 6257
ai_edge_torch/generative/examples/paligemma/verify.py sha256=KT3Ruy40tSESxQuy-Sw01NAI3zId1BZr6Bp7FZj1wZk 5622
ai_edge_torch/generative/examples/paligemma/verify_decoder.py sha256=al5wMPWri4IRVWrLmCplPi6uoCzwh0vBHMGnCt-XUqo 2690
ai_edge_torch/generative/examples/paligemma/verify_decoder2.py sha256=tm-UfLr0YeBRVcQsWLBOMWI9JUzHmtPEbYK2vpITpqY 2534
ai_edge_torch/generative/examples/paligemma/verify_image_encoder.py sha256=vNm-wTT8BD6zbX6GocfP1QrVoHl0zSvuVxoXN36eeiU 3540
ai_edge_torch/generative/examples/phi/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/phi/convert_phi3_to_tflite.py sha256=cD8rtwgYeGrXB9sYVV_D1AB8Up1AWNS-1XtrRlyzE5o 2296
ai_edge_torch/generative/examples/phi/convert_to_tflite.py sha256=G1i_ybDCTBaOD1OOCTk6jqOf__xYYZvhXcxY8MXhPHw 2294
ai_edge_torch/generative/examples/phi/phi2.py sha256=c6PYCky7yJn6MVIYOCTx8S_CH27kOPmJbRZcI95nbZs 3477
ai_edge_torch/generative/examples/phi/phi3.py sha256=7Y1E4XpRuZOiSbeZJ-C2uJjmlnDtWv6L0XvPRE8oEQs 7112
ai_edge_torch/generative/examples/phi/verify.py sha256=YPFCdbnfmvq38fbpBNr0kHPfSZo4p3_6WkLJAW3pLPo 2177
ai_edge_torch/generative/examples/phi/verify_phi3.py sha256=kVYaBVvddfQng0IyZGxyTJEzhiPO0G4VFJm2WOc2Q94 2360
ai_edge_torch/generative/examples/qwen/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/qwen/convert_to_tflite.py sha256=BHkDsivbbfVBPxknkgWwtLskvxyrd42TXuCB0aLVbMY 2633
ai_edge_torch/generative/examples/qwen/qwen.py sha256=Zi_qiQ1JPokXZ95jgSEnQp3F-LKzFCvWvFLKhJjnASo 4199
ai_edge_torch/generative/examples/qwen/verify.py sha256=9_AyEJTeUfvhhID64Rto2bflFPyXMFokdQLsseLUMiI 2775
ai_edge_torch/generative/examples/smollm/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/smollm/convert_to_tflite.py sha256=RKmSBMrup5A2bsPPaTdrBQb8NeRiUHy_1SUOA8DAs9U 2305
ai_edge_torch/generative/examples/smollm/smollm.py sha256=kk3cB_qaCzbFOhHtJlLb7qvSEBQTsILnoAcSFE3AkpE 2711
ai_edge_torch/generative/examples/smollm/verify.py sha256=HXYcCjDJMylVL3Pc9HU-UXqtpjtIU25o1YhPiX30aPU 2361
ai_edge_torch/generative/examples/stable_diffusion/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/stable_diffusion/attention.py sha256=kDWG6MlIGa89zC5KSRcJlw2c4ITuw8KcchtfmF55f4g 3545
ai_edge_torch/generative/examples/stable_diffusion/clip.py sha256=5M4auM33SgCTODt0VT8TO-EVILruqGDRiNILBPeB83Y 6072
ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py sha256=sB_7-PVri8PxKnFG7c8GsTGyrxGEda-oZwGyyScTL3o 5239
ai_edge_torch/generative/examples/stable_diffusion/decoder.py sha256=sQKQ-k6H9kG2brgwLsktjCMeN2h0POyfMP6iNsPNKWc 16271
ai_edge_torch/generative/examples/stable_diffusion/diffusion.py sha256=6W58LxmHHkz2ctgpknQkyoDANZAnE9Byp_svfqLpQf0 34793
ai_edge_torch/generative/examples/stable_diffusion/encoder.py sha256=CAPsW84A8f00nS6fLFeh_XUjCPsDCA5UxHOUsMrLfSU 3450
ai_edge_torch/generative/examples/stable_diffusion/pipeline.py sha256=GnY3vPZ-obrWuJifuE5bUooKLqAI7v6q71oaTuLKeBE 8778
ai_edge_torch/generative/examples/stable_diffusion/tokenizer.py sha256=xychak9hdLd6ieXBYEwrK2BkF8NRZWZSSCijIsESpBA 3420
ai_edge_torch/generative/examples/stable_diffusion/util.py sha256=XIXIB0vCvQKOGyIyiZeiIA5DLeSXjkudywvJS4FK7AM 2431
ai_edge_torch/generative/examples/stable_diffusion/samplers/__init__.py sha256=uQWKzCD_49ackNFrt50H04dkDXxfAwUCtMWWQre5SVE 830
ai_edge_torch/generative/examples/stable_diffusion/samplers/k_euler.py sha256=wBBNM24waZ57M1rXonwesfUkKe9DqpqO3eW6BfZkrD0 2323
ai_edge_torch/generative/examples/stable_diffusion/samplers/k_euler_ancestral.py sha256=c89ldwtuQ2_yspGrGa7oh7fsvTt6A86Whxa6fBK9YOQ 2526
ai_edge_torch/generative/examples/stable_diffusion/samplers/k_lms.py sha256=ZE6HyOoBJrmTh54KVFf7DjNBnBS0pT4cgviYaq8HGMU 2801
ai_edge_torch/generative/examples/stable_diffusion/samplers/sampler.py sha256=RxR5rw0wFFm_5CfAY-3-EIz83vhM9EKye8Bb5zBb0Ok 1341
ai_edge_torch/generative/examples/t5/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/t5/convert_to_tflite.py sha256=HHtZTtUh3QgE4F74-ru_8n1pt6cqfbObw12xoaMJ7NQ 4596
ai_edge_torch/generative/examples/t5/t5.py sha256=gFTmPi-xB8pcPRgoF3DJxvH_fT-KWTb8ii77P5UbKR0 21263
ai_edge_torch/generative/examples/t5/t5_attention.py sha256=l01oYyJo77INzRwN4xqXquaFQPvCFBFF5zOnmGVb3Hg 8731
ai_edge_torch/generative/examples/test_models/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/test_models/convert_toy_model.py sha256=6-WaNHckq_LlXMVTh8x90MGWeWq2bu_T_XQd3w9FnGg 3261
ai_edge_torch/generative/examples/test_models/toy_model.py sha256=4113jZK-Hu3kYop__WTc8Bq-bG6YzQtADbxHtYPEB4w 5036
ai_edge_torch/generative/examples/test_models/toy_model_with_kv_cache.py sha256=C9dzJFK3TybxKpM1vSdLjOKftkJ72DGjr8YR4H7vCe8 4664
ai_edge_torch/generative/examples/tiny_llama/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/examples/tiny_llama/convert_to_tflite.py sha256=5rgbTIxHoFg8sTnzrGA_ekT-HJEt9p7Dla7cIY874jU 2338
ai_edge_torch/generative/examples/tiny_llama/tiny_llama.py sha256=mhJ18rb9sxrYRzv1YSzhbNs97oUZck99avZDcUO2oV8 2800
ai_edge_torch/generative/examples/tiny_llama/verify.py sha256=7Bk8z033M-BCXJ299fpQNXYAudBbZoDQp9934xcvg50 2426
ai_edge_torch/generative/fx_passes/__init__.py sha256=jrzCB3ZyY_t5jJM1e2Czdt3DjAIL43R0_a-T-I7wOzw 1155
ai_edge_torch/generative/fx_passes/remove_sdpa_zero_mask_pass.py sha256=hhxSQvkDMv0isZJhmuLiod66ZODaJ8uSPSVTJVHBabQ 1931
ai_edge_torch/generative/layers/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/layers/attention.py sha256=_OmamS3f0m_JtW73ljwGLwFPeMLL837JCLY-dJ3iRUg 12453
ai_edge_torch/generative/layers/attention_utils.py sha256=zBVwlBUTs-nStIKCZG0ks5ra7tsqc9ShfakFJKH5rds 7344
ai_edge_torch/generative/layers/builder.py sha256=LXGuSHIx6QZAzLFm7aJvlzoMPgQwbXLFchGEKYwOOUA 5090
ai_edge_torch/generative/layers/feed_forward.py sha256=hdICat-8gW7-vxDAevJQ8NQ-mynllPiqLdXQMF6JMnc 4189
ai_edge_torch/generative/layers/kv_cache.py sha256=DhHIggaOQ2IAY4aRuMAuCLWZv1dBz5PYtmOEjkx9EQY 6291
ai_edge_torch/generative/layers/model_config.py sha256=viX51T_naJ9sPpPxPoMnSueBPYE2zxWNOD0xn0f-_bM 7510
ai_edge_torch/generative/layers/normalization.py sha256=h2btgRHMMjOcyLm8adEmcT0pG6imq4QcWblKJK5MYXA 7479
ai_edge_torch/generative/layers/rotary_position_embedding.py sha256=zbFTNgQdOT-tcKK1QaIX6fG-50syYwQX_ZbLhg2C98c 2691
ai_edge_torch/generative/layers/scaled_dot_product_attention.py sha256=gXxh3papKy4FBpGEX7VyZ7rZ1Js6aHK70Q6DKrVSckY 4154
ai_edge_torch/generative/layers/unet/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/layers/unet/blocks_2d.py sha256=9jKzOfiBQ66bp1ZnVIAoREIifVNFx4aTlQeYMAx2_pA 29062
ai_edge_torch/generative/layers/unet/builder.py sha256=zAqWXdimmMrQRhmE_t9XkS68mh6PSrzwb-2NZZXrR5I 1901
ai_edge_torch/generative/layers/unet/model_config.py sha256=pPDwLawc23pfMaPVyMJlYmxVVusjMvx-l8wBwOYOH-c 9692
ai_edge_torch/generative/quantize/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/quantize/example.py sha256=1lfVNUd2cEyRUnoZ7BLbRJ9IN-FTKiWBtZNPFUzAiWE 1747
ai_edge_torch/generative/quantize/quant_attrs.py sha256=n1Fm8BFC8gJa_oiwwAOOghJyHtOXYZ4q-5ZRy4pHrIw 1957
ai_edge_torch/generative/quantize/quant_recipe.py sha256=tKnuJq6hPD23JPCB9nPAlE1UHAwdbChkgPShiVaz4CE 5156
ai_edge_torch/generative/quantize/quant_recipe_utils.py sha256=4fgmP_GgeiFUOkIaC9ZZXC12eO3DQZdrWDXRz5YXiwU 2270
ai_edge_torch/generative/quantize/quant_recipes.py sha256=0Kvr_o7pbMnE8VMe6Ml0FBxkHM6RJ3C14B2I1mjItjc 2030
ai_edge_torch/generative/quantize/supported_schemes.py sha256=FjdycEOvxRgBmQdZVufetPvkDoD7rUowIOSKV9oV5Kk 1418
ai_edge_torch/generative/test/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/generative/test/test_custom_dus.py sha256=gxG78CcTpXF3iLzDR15Rlz1ey1tNTlSdkp6TeYEijp0 3301
ai_edge_torch/generative/test/test_kv_cache.py sha256=2AulHBS3hC4b_68PNNBkRVOrypy4IM5YjC4p-6dgCMM 3793
ai_edge_torch/generative/test/test_loader.py sha256=9mQUeeZKOVApOWSWl2cN9c10axZjMKM1-0Zd823CCS4 3449
ai_edge_torch/generative/test/test_model_conversion.py sha256=jfqkECCX7XKHeBAuDXrkwQJf0vM72eG3LMc5rluha84 6191
ai_edge_torch/generative/test/test_model_conversion_large.py sha256=NctnggTSFh0XEQbTu55diZ35rFD2QIARO-8PzLktRWg 12165
ai_edge_torch/generative/test/test_quantize.py sha256=bEJMhpQ9bIDUZVBXTW888728FcH-i3SyE4JSZZUgU0A 6071
ai_edge_torch/generative/test/utils.py sha256=tF6aCfAGJnc9dmzCnZCEOuKNVimfWOqscv9og0DDLHU 2656
ai_edge_torch/generative/utilities/__init__.py sha256=-_jxnnFnCgnTU4oTm4MnRsvL5lqhomBNdFBbqfmfHPo 720
ai_edge_torch/generative/utilities/converter.py sha256=hIwWUWjgPvWLATtsYYG6RWbFQWhOr2RpPlMrd-4Am9U 5959
ai_edge_torch/generative/utilities/dynamic_update_slice.py sha256=e2mhx-Vp8sUK4EXoPtpZLSx3TViqLAKs67EhKcXBjAQ 2121
ai_edge_torch/generative/utilities/loader.py sha256=A3SOjPXp--AsvoP1hqj5QKWE4sgxoFc3H5EBUz_Eogc 13531
ai_edge_torch/generative/utilities/model_builder.py sha256=S08WNqVKCmxd2QjtMlwETd7J97UnlME_bTKdz5LMkGU 6352
ai_edge_torch/generative/utilities/moonshine_loader.py sha256=_RpFabSqtGH5PHiP3_1f6QfO14qMADUxr_HGRlVDFB0 4891
ai_edge_torch/generative/utilities/stable_diffusion_loader.py sha256=dqPD9qRXEWtU3ombslOC-BE2l_dMwHoCNu7NsIJhsso 36158
ai_edge_torch/generative/utilities/t5_loader.py sha256=tEsfy8-ymzbbjOIc-oesXF3yGyyWtJgFXn2s7VOavt8 16961
ai_edge_torch/generative/utilities/transformers_verifier.py sha256=8sp9m_FMcXn7nqOrochtu2jIANkJKhnhIBUmH0ZTDR4 1549
ai_edge_torch/generative/utilities/verifier.py sha256=awO-sQrEpsFxIkZw72ysWZenYEmkLOLOuj62o2c7XeQ 11994
ai_edge_torch/hlfb/__init__.py sha256=sH4um75na-O8tzxN6chFyp6Y4xnexsE7kUQpZySv6dE 735
ai_edge_torch/hlfb/mark_pattern/__init__.py sha256=cjTprggj_cuktSCm7-A25e7Shop3k63ylp7sdZmtZ8o 4790
ai_edge_torch/hlfb/mark_pattern/passes.py sha256=pjkKcI1nHECPluAt87cFBrt1DP0f3ge7rHq1NhCkBIE 1936
ai_edge_torch/hlfb/mark_pattern/pattern.py sha256=NP2mYhe5D2GjtqQfqqldp-ko3xtNghuFKKJOQskUJFI 10041
ai_edge_torch/hlfb/test/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/hlfb/test/test_mark_pattern.py sha256=ivq0eVjuf31idfNY0E12F4FxdkSI9hwYXapLJBkIf8Q 4831
ai_edge_torch/lowertools/__init__.py sha256=A8WBXvWtuFYYWtNTqPD7waVntLaSVAnSMwx5ugjZBIw 761
ai_edge_torch/lowertools/_shim.py sha256=Mbg16tnCVK0YsHowfbpqpNX1qySuMLvpGI_-I5SIrG0 3276
ai_edge_torch/lowertools/common_utils.py sha256=Z7p-ivOHtddktpnHrlDm_dSoTxJOdEjFXIGQbzjgwQo 4504
ai_edge_torch/lowertools/odml_torch_utils.py sha256=dxg2pBuVhSZeY2Ouc0F6nHiZilWZmpSPA7I8kGqSkVI 8282
ai_edge_torch/lowertools/test_utils.py sha256=mdxTlhqHABZEQ_GEmPFCL8LIAWtqRtYZUGdSY1ieZjw 1949
ai_edge_torch/lowertools/torch_xla_utils.py sha256=tH5BW8-Up1uy5Iq1LdXiJInXBh4-YqNXJpSwwy3kwSg 9460
ai_edge_torch/lowertools/translate_recipe.py sha256=ymkBpFqAUiupRWqrPOWiVphKcXR1K5vHK0RjgBFtxlE 5652
ai_edge_torch/odml_torch/__init__.py sha256=S8jOzE9nLof-6es3XDiGJRN-9H_XTxsVm9dE7lD3RWo 812
ai_edge_torch/odml_torch/_torch_future.py sha256=AJ0klpsbu2ZBTfiZlqSOoaYzBVITt40a1fYN8xKkEPw 3044
ai_edge_torch/odml_torch/_torch_library.py sha256=Lw1gqL2HWNRspdTwNhIkYAHDyafHedHtkXyKKxn-Wss 805
ai_edge_torch/odml_torch/export.py sha256=QzOPmcNPB7R-KhhPEP0oGVbDRgGPptIxRSoz3S8py9I 13405
ai_edge_torch/odml_torch/export_utils.py sha256=QeA37Irlty6AiIBuqmHmJgn3lqahBQ5xsh6IKRoKm1g 4774
ai_edge_torch/odml_torch/tf_integration.py sha256=lTFJPPEijLPFmn6qq2jbpVTQOo0YaOTK36kK6rCiyIE 5956
ai_edge_torch/odml_torch/composite/__init__.py sha256=71GM_gDZxJyo38ZSoYSwhZX3xKA9rknO93JS9kw9w_c 778
ai_edge_torch/odml_torch/composite/mark_tensor.py sha256=U--rwl-XkWKgkdXCXDn6yySug8FR66o1YFUAIoSaWW4 3523
ai_edge_torch/odml_torch/composite/stablehlo_composite_builder.py sha256=2Y52E_gLeoXpMcPpV-svXsgN3JbEIjnPVjm0xkpTUdQ 3319
ai_edge_torch/odml_torch/debuginfo/__init__.py sha256=9ag6-WWRG50rPCtIV7OpIokEKu2YRyGlMZZqVPWUH6g 762
ai_edge_torch/odml_torch/debuginfo/_build.py sha256=1xCXOs3-9UcsOyLFH0uyQwLu7c06iYFTo0NQ7Ckbl2I 1465
ai_edge_torch/odml_torch/debuginfo/_op_polyfill.py sha256=IvOBQyROI9WHS3umHRxsDW-1YElU9BPWzKtJA2eKWOI 1739
ai_edge_torch/odml_torch/jax_bridge/__init__.py sha256=e9Oa4J3An9FYr3zM0OzjzyNNitEeg-IoBUGNfUxsDSA 798
ai_edge_torch/odml_torch/jax_bridge/_wrap.py sha256=LqwZ1vCJTSOzgzvH8LUAN-sAkF-l_pGj1AMEIzAqHCA 6638
ai_edge_torch/odml_torch/jax_bridge/utils.py sha256=T8isGc896VrHZ6c_L5pYmLpolQ7ibcOlgWfPuVFPzIg 2264
ai_edge_torch/odml_torch/lowerings/__init__.py sha256=GWFl7WWgExLXu6FEYxnig5_g6hd_Sfnl8690uFg2-CU 1013
ai_edge_torch/odml_torch/lowerings/_basic.py sha256=8mZTp_ybcMO3tDRQdlDP68BVeTw560XsTR4XH-ldTdc 9987
ai_edge_torch/odml_torch/lowerings/_batch_norm.py sha256=PaLI0BB6pdBW1VyfW8VTOT_Be-ZcqYdNOsyfzKfq8Cg 2064
ai_edge_torch/odml_torch/lowerings/_convolution.py sha256=Q0aDzyUcZMoSzSbOU-r3LJMgPe6fble0QwdYVIOHHHk 6887
ai_edge_torch/odml_torch/lowerings/_jax_lowerings.py sha256=CJHWkmY4aAVQ5dmFsVc3Ox9TPkoLSNOfa96psD4CLRo 11561
ai_edge_torch/odml_torch/lowerings/_layer_norm.py sha256=khJIvDVk2s332Nd2Be-5dM6-wp5DGff61HCV5lskHmQ 3011
ai_edge_torch/odml_torch/lowerings/_quantized_decomposed.py sha256=XDZ0zLej_XaQDJnaAAxhNFAd7NfQm5SOVEp_nno_krA 6178
ai_edge_torch/odml_torch/lowerings/_rand.py sha256=g6SuqDkuC6hD35lyP1-5H7ASDIzPSmKukeNT5naZSv8 4133
ai_edge_torch/odml_torch/lowerings/context.py sha256=jslcCv7r_HtImSRTxJwHAUV_QCu9Jub51lovmoBkmFA 1295
ai_edge_torch/odml_torch/lowerings/decomp.py sha256=UoJeZVcr4zAN_11i-HzfOhxGCxUm-7b1JXPVBxR2hSs 2414
ai_edge_torch/odml_torch/lowerings/registry.py sha256=Tp2h11l5uTifO0aIkuUOWAF_ibEjmd65Xx99w3EXuGE 1924
ai_edge_torch/odml_torch/lowerings/utils.py sha256=pqM6mumpviFDHRaabp93CUAngzEZmWcAHl0nTDgyI2g 6167
ai_edge_torch/odml_torch/passes/__init__.py sha256=AVwIwUTMx7rXacKjGy4kwrtMd3XB2v_ncdc40KOjUqQ 1245
ai_edge_torch/quantize/__init__.py sha256=aB5dXot04bqyUhpsDFvxt9CIi15QAC4euvqOndJ0XLU 714
ai_edge_torch/quantize/pt2e_quantizer.py sha256=CKIEhs9jCcna64qj1jFH9zEbMbRdyeGV_TmSqEBPjes 15741
ai_edge_torch/quantize/pt2e_quantizer_utils.py sha256=eARD1LxLi5m7Z0n_psAkeX_AtUp4fNkE--oECBfivv4 36208
ai_edge_torch/quantize/quant_config.py sha256=U0KisSW-uZkoMJcy-ZP9W57p3tsa594fr9PphCRdO8o 3172
ai_edge_torch/testing/__init__.py sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g 671
ai_edge_torch/testing/model_coverage/__init__.py sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE 765
ai_edge_torch/testing/model_coverage/model_coverage.py sha256=UPB448aMDUyC0HNYVqio2rcJPnDN0tBQMP08J6vPYew 4718
ai_edge_torch_nightly-0.3.0.dev20241224.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
ai_edge_torch_nightly-0.3.0.dev20241224.dist-info/METADATA sha256=J8nAtbMNmSIcHuThVv0omkhpldggz91pIIYy-6ATJgM 1966
ai_edge_torch_nightly-0.3.0.dev20241224.dist-info/WHEEL sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A 92
ai_edge_torch_nightly-0.3.0.dev20241224.dist-info/top_level.txt sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE 14
ai_edge_torch_nightly-0.3.0.dev20241224.dist-info/RECORD — —

top_level.txt

ai_edge_torch