lightning-thunder

View on PyPIReverse Dependencies (1)

0.1.0 lightning_thunder-0.1.0-py3-none-any.whl

Wheel Details

Project: lightning-thunder
Version: 0.1.0
Filename: lightning_thunder-0.1.0-py3-none-any.whl
Download: [link]
Size: 483953
MD5: cab06adf92a0c61cd36701c5fda0e2ae
SHA256: dd95bbfc2fbd81a1e8cafc34d28ed94b33ae77d6656c617887af3dc1091cd550
Uploaded: 2024-03-20 19:19:05 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: lightning-thunder
Version: 0.1.0
Summary: Lightning Thunder project.
Author: Lightning-AI et al
Author-Email: community[at]lightning.ai
Home-Page: https://github.com/Lightning-AI/lightning-thunder
Download-Url: https://github.com/Lightning-AI/lightning-thunder
Project-Url: Bug Tracker, https://github.com/Lightning-AI/lightning-thunder/issues
Project-Url: Documentation, https://lightning-thunder.rtfd.io/en/latest/
Project-Url: Source Code, https://github.com/Lightning-AI/lightning-thunder
License: Apache 2.0
Keywords: deep learning,AI
Classifier: Environment :: Console
Classifier: Natural Language :: English
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Scientific/Engineering :: Information Analysis
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Requires-Python: >=3.10, <3.12
Requires-Dist: torch (>=2.2.0)
Requires-Dist: looseversion (==1.3.0)
Requires-Dist: lightning-utilities (>=0.7.0)
Requires-Dist: numpy (<2,>=1.23.0)
Requires-Dist: igraph (>=0.10.4)
Requires-Dist: optree (>=0.9.2)
Requires-Dist: opt-einsum (>=3.3.0)
Requires-Dist: mpmath (<1.4.0)
Requires-Dist: ipython[all] (==8.22.1); extra == "notebooks"
Requires-Dist: absl-py; extra == "test"
Requires-Dist: coverage (==7.4.3); extra == "test"
Requires-Dist: einops; extra == "test"
Requires-Dist: expecttest (==0.2.1); extra == "test"
Requires-Dist: fdm (==0.4.1); extra == "test"
Requires-Dist: graphviz (==0.20.1); extra == "test"
Requires-Dist: hypothesis (==6.98.15); extra == "test"
Requires-Dist: jsonargparse; extra == "test"
Requires-Dist: numpy; extra == "test"
Requires-Dist: pandas; extra == "test"
Requires-Dist: pytest-cov (==4.1.0); extra == "test"
Requires-Dist: pytest-random-order (==1.1.1); extra == "test"
Requires-Dist: pytest-timeout (==2.2.0); extra == "test"
Requires-Dist: pytest-timestamper (==0.0.9); extra == "test"
Requires-Dist: pytest-xdist (==3.5.0); extra == "test"
Requires-Dist: pytest (==8.0.2); extra == "test"
Requires-Dist: xlsxwriter; extra == "test"
Requires-Dist: jax; (sys_platform == "linux" or sys_platform == "darwin") and extra == "test"
Requires-Dist: jaxlib; (sys_platform == "linux" or sys_platform == "darwin") and extra == "test"
Provides-Extra: notebooks
Provides-Extra: test
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 6405 characters]

WHEEL

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

RECORD

Path Digest Size
thunder/__about__.py sha256=YgyjKshgRYvGit_yHjqco7TqkTjyniz_MHsn1LJLYbs 424
thunder/__init__.py sha256=8R04yuOfm4vYMM3AbB97w-sHjmzsx0MhGjPTwWZMmLM 32320
thunder/common.py sha256=FPwLrE3KBcCjaS8xzXmCWvrh-D5KBr9q9kmcQraKu5A 31862
thunder/functional.py sha256=BVxjCNAXGz8ZLaUBD1_bUyhBktm4CUFli8vOk_YW1Uc 16957
thunder/benchmarks/__init__.py sha256=I9SqsK3LLqsbtF7t_Z_kLCGTZybs92yhE8VUiSAG8Ik 101413
thunder/benchmarks/benchmark_litgpt.py sha256=HxVNqZXSy_STMVCJEZ_VDHSXiaA5Gnq-PNCN6oh_rvA 18879
thunder/benchmarks/distributed.py sha256=umxI0xwrqey1lTgId8246yx--GBSxymfzBbr4LL9tp0 19856
thunder/benchmarks/einsum.py sha256=TxaxK-vErvgdnt4JODatV6MfCkmfuf8sOFlM4mfq0xY 3726
thunder/benchmarks/targets.py sha256=-_d7tDBW_o388jA6vBIm439a8gEt12_-sJWN-Y_JwuM 26229
thunder/clang/__init__.py sha256=0pV8OQSeDEbBtSGLe_MtArJ2MBp9z1JBq7-NkBYOT3Y 62876
thunder/clang/langctx.py sha256=yyyQWi7zPhRZlBZ0B-nRZDhtY6o1fEc9U_uD9R3iJJI 2516
thunder/core/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
thunder/core/baseutils.py sha256=WtCxLfchoL1cOF1HzECibX9kjBAjLBuxH9vJ5O-QMBo 14999
thunder/core/codeutils.py sha256=PU8IGQuFRFLda3qfkJwtU5l1m2CqZKIX2NaCuBLJTZo 13335
thunder/core/compile_data.py sha256=N6-8BVRGWvAdcqb4L4R0YoUmSE8mztiYxmfZ1QmvP48 2139
thunder/core/devices.py sha256=1Csfe0Gldq_aWS7_vs2xVQg6y4nyIHBRCkxUxlqnC78 5921
thunder/core/dtypes.py sha256=XZjVRG1JZR7_gu9iqw1htGVqSCJ-B90jeOpt98AusEk 17106
thunder/core/interpreter.py sha256=wtKIhsEe_U-sj7nUc26mEn2b0gkG42UQTwChFH4cAlE 252847
thunder/core/jit_ext.py sha256=YyRD0uHJoKtTlpzCJu7oW8h0HRv0joMc_qb9f7Uen7E 52096
thunder/core/langctxs.py sha256=BVQmKXjQkQHHOGjMOpR9_jb97ck2Ask1Hy4w3GCnups 4146
thunder/core/options.py sha256=eqtkbWdNDRK7TFiRHkazBA8rF6Asvj-5JQGRkGvHTR0 7046
thunder/core/patterns.py sha256=YK6UdmPr8AHY3KKETcmS4iMwXas1sXNIybRu6_wl45w 14058
thunder/core/prims.py sha256=Q4kpoDfqoz28B44ym39gaqYLebpKQrvzFY-gKf6vcsM 112701
thunder/core/profile.py sha256=HLRypcCMdpBKjWBL4-mnXnlCF8cON8BUu42pQDRkXKM 629
thunder/core/proxies.py sha256=K5DdAFA7D8loUONNOXndCzDXSKrxXiMiVZGunFiv6a0 48752
thunder/core/pytree.py sha256=o8FlJiaWoCcu5cZqd3Ssqkd23OI_h9GsRsFoRzuOHZo 731
thunder/core/rematerialization.py sha256=kDZatfoUgQ4VxJLmNOCh1TYEjyDAj82tMs68bBXgVGk 28140
thunder/core/symbol.py sha256=H5jUtauY-PF6J5u-fyQTgO2EJxHb07n-4eaZKfo7IUQ 25222
thunder/core/trace.py sha256=AvwpnFUigewdIzW7J3EvCgA8o2heUaJ-Z-ZioYPYrrU 18726
thunder/core/transform_common.py sha256=rINFM_oh8N7mevhW1dMwas4PRj13Ebi99TH1LWm-7tg 10446
thunder/core/transforms.py sha256=18qi8IRJ2gqxK-JAg59M6S6DCTrE-VmkIqLXOrQFbMQ 141740
thunder/core/utils.py sha256=0-w7alEdRZFwikWnSL4md_bTqkEDX_go-zQ4nCacK-I 37038
thunder/core/vjp_utils.py sha256=oep8Cu5gleYHnf2PLciCHuD-pR15erSbPOZRAXSdCIs 7373
thunder/core/script/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
thunder/core/script/algorithms.py sha256=xZVMMcVcxixfZf-qGCvO5nh6AwtTk8l01FXgu6JDBX0 8642
thunder/core/script/frontend.py sha256=i2QbTzKGmXVW52uNvNZHq2RHDdwMjxGNzNDExpQo_1g 28000
thunder/core/script/graph.py sha256=3qRE_gsAZTFx-Z_OZR9Yp3zDihUN3A2eX3L6jZ0cpYM 31183
thunder/core/script/instrumentation.py sha256=Pm3D_jdC5ZXvXbxsczORA3YymtxXanFXCMbjgwqHVj0 4333
thunder/core/script/noinline.py sha256=ysOLxzZizHPXDOt_hyCxlMFX9tD_h9-VCbCtx0GO3Uk 833
thunder/core/script/passes.py sha256=aX16r6Sx3HVshymxKrZG6x_84enlxpGoy35zcyTIeXk 38591
thunder/core/script/protograph.py sha256=zzFCkNXnfxvch2-hZshE7X5lPye84yboJALhzsHZicc 23974
thunder/core/script/protograph_passes.py sha256=pP7RRDvU98izGBi7Fw2kC5CYzQ7eWNiMhGYNweAEFEQ 3488
thunder/core/script/python_ir.py sha256=0pDlDq-8V77ZEbHH_bicN_wRw4VStnhIgttjJZGkgKc 20335
thunder/core/script/python_ir_data.py sha256=FRXE-V35EquOL8W0piEMONbpqjv2l2VP_kEoHEVVTm4 1891
thunder/core/script/parse/__init__.py sha256=nL0QEdSt2EdgGxj_HNPbylGqyOx1WKLBJ982nx--E4I 319
thunder/core/script/parse/disassemble.py sha256=IiZYkIG9MTLXLcJZ5YUdm7rr1HXwbZRRItx8AFVZFds 6717
thunder/core/script/parse/functionalize.py sha256=CKIg3vGytpT87_xpCl8-vM6b7PH64_YlL8zfdwilLYw 13559
thunder/core/script/parse/instructions.py sha256=FrtHGrR2zRd9Px-GBs6K2eKyZA9Xk97NVMdd3MsuIS8 3796
thunder/core/script/parse/stack_effect.py sha256=CBxgNmfbwEElLAfg2xMKQfRh5sUqekEvHbgPS2kaMpM 9168
thunder/core/script/values/__init__.py sha256=PEXMzd0om9C1hVO2lSwAsQY3NU0ynAAtS1JwfiFEF20 204
thunder/core/script/values/base.py sha256=eQ-9n6hEQtW6cs68g1aBgNP_kQIpPFMt12AqvcjGStk 6401
thunder/core/script/values/composite.py sha256=VFMj52U__hSChXzLPX9S6lMrqyEaP1_GAa2OjTEsIuo 7471
thunder/core/script/values/materialization.py sha256=eIqM0hTe-AZf53aJrTE3BS2lCwQZJ9MXwEid7tZDIEA 7777
thunder/core/script/values/symbolic.py sha256=C2cAuH0tiItbgJwojkbCq24vY9-nWC7avH3Kz1wTt58 7788
thunder/cudagraphs/__init__.py sha256=32uWI5PO7zWwNG98fvzkhIdViDT2W7eKizpSXtEGWRM 5265
thunder/distributed/__init__.py sha256=yibXgu2ojqscjILtIKoStSJBE2jUyIuyNeapT2kVTiU 16803
thunder/distributed/bucketing.py sha256=vFa8TkbsHHzervcwMKOlydga1l5-Epnp4Gj9KtE5T_I 7101
thunder/distributed/checkpoint.py sha256=RY2omSjzTOFphMDxpnDescszDfVkwJ3qfhPMpOAPsRA 9307
thunder/distributed/prims.py sha256=lYInu5ROoSRqasb9sg1dcRJFJtjMeNArYJrTqGII9_w 10906
thunder/distributed/utils.py sha256=gEYgU8OzQY1a12UVvC2hQBHBLFsIFko7UX96FbyZ11g 10907
thunder/distributed/transforms/__init__.py sha256=2KgafzTRe3b97mZjkMQMNBZiTsfHg5zQB-IGPz8NUTU 171
thunder/distributed/transforms/ddp.py sha256=gAVxDubOr2nVzLIdTz5npP4YVpEfMc_g1lERov7eGaQ 8449
thunder/distributed/transforms/fsdp.py sha256=LcHSUnon32spWTNoLSUEGu1j9NfpLeRS9UkAR1N3Ui0 28416
thunder/examine/__init__.py sha256=Id6I058YVTpxuF7S_YcLamWGILDpRQofC1dx66bNlvY 8681
thunder/examine/memory_caculation.py sha256=vnlYRFZ5blXjcDgId4CO_uW0QzsgqjLl_IU3MIckd9g 5761
thunder/executors/__init__.py sha256=OwkR4gJYYE6MhVan751f-k29bZU5GG3M5N51ws9ZR2I 598
thunder/executors/apex_entropyex.py sha256=PvWAFTwQLnh6xzPeTfQTyyOLdZAcu5YTQ0ne-EL2kvs 9365
thunder/executors/cudnn_layernormex.py sha256=mQe-MMT2l11Xqf_xnKpGQgs7uB7ZEHUPHyC5zy4iOsk 4500
thunder/executors/cudnnex.py sha256=smLq-IrZ1PXVa9xlsCYffhI0FAFZCjpj2sZSJsSL4M4 22634
thunder/executors/data_dependent_partition.py sha256=NmC27DmlLPpxVVlGMoMibWr2cHffS_wlBi-AKgG98wM 10546
thunder/executors/nvfuserex.py sha256=dli33376k-VepDjSEmOrm3rhHGiFZFE8HL7fPEyy5EA 1085
thunder/executors/nvfuserex_impl.py sha256=6uIWnXLTst8ZsPAnrBK6nREWWGUb9E0XBXiZEGtBtyY 73540
thunder/executors/passes.py sha256=8OB_zLRlIFY_PthdyC36qcZ4qHmipcobDVAAHlCknUY 10861
thunder/executors/pythonex.py sha256=TO2ztPAdClzGVLARUUxVBSlu2Leau0T4nQnsMZtPCxA 13938
thunder/executors/sdpaex.py sha256=XYey5Tg7rF_HkYZ8bi5ZQIylxptHUNklnTjAlsjhKVM 26538
thunder/executors/torch_autograd.py sha256=YrUqYQzMbkdzTxyzKCSz3jWtTp5OfVg8k0GHFdWela8 14969
thunder/executors/torch_compile.py sha256=3nyEIXAp7BBrmrvZLcpk3vUCbx5SRx_c8DoP24lDy8s 7908
thunder/executors/torchex.py sha256=opAWiGOwhstcIq1Wz4IzHTVh2qwPxk-DzBxupk9nY88 75141
thunder/executors/transformer_engineex.py sha256=aAih126AbTOltpTKSsxv9TyLWXdOTtHGDy4E7lCdu3A 17107
thunder/executors/triton_crossentropy.py sha256=_-GeA0Xt2oLsVBsHF5zJavAYwEn-w-EErYBd6I4XX7s 19808
thunder/executors/triton_utils.py sha256=eqMFeBJKiA0Jm4OIiJEcfCgcmlqzt2XKNJLZduzs2kg 443
thunder/executors/utils.py sha256=GnrrO90_1UGwrSzPJLg4eAEeyVilnrTZOlXOqOuLrF0 2646
thunder/extend/__init__.py sha256=QlZIythlQNtvce0_1sIUELCqgwt9zRMIro1EuWmhmnU 12091
thunder/numpy/__init__.py sha256=1YqjYDRI_P1jvl8es9QQ5tc93qomtar56C7UzfYdLxs 1652
thunder/numpy/langctx.py sha256=4pMXSJ0EYpucCIPHDfDQCaA7K27aiHs8N6odArnLAfs 2554
thunder/torch/__init__.py sha256=xrcNtv1GkW4fMtd5j35EReVwLF8SX9hHFBImUTPZFso 142813
thunder/torch/langctx.py sha256=c68UWNSh1f13M6QAdkpvhcYIvK0IcZbf_Ds-Nd44l98 2494
lightning_thunder-0.1.0.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
lightning_thunder-0.1.0.dist-info/METADATA sha256=88pabamNDsmql_2iJJaHU29Kt6Sj2BDe0eea7Qmd1VA 8975
lightning_thunder-0.1.0.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
lightning_thunder-0.1.0.dist-info/top_level.txt sha256=H3Kmgbx7mW8sBDn18vcmMUqB_NO4MO0EddocH6RLuMM 8
lightning_thunder-0.1.0.dist-info/RECORD

top_level.txt

thunder