spectrax/__init__.py,sha256=f1f38f84c2040d321fd677ccfee130dbf9e870c1d127baf656c1bed65e2e0cc2,4615
spectrax/_version.py,sha256=003860786e85221cbf03c0b6751d6952452c666d7a258b67d6109958e7d222a1,761
spectrax/api.py,sha256=5e6bcb902546813bdd62ae63571f61a1007873fdd3adaedf21f3008af967d071,12481
spectrax/contrib/__init__.py,sha256=0f5b6523674dca96558d17259f61e920c193cf21e4ffa31da11055e7e825badd,888
spectrax/contrib/optimizer.py,sha256=3151a2e67b89256e3a7ed19433ea58dbc27c7ad4f1011af8e468032f8bf2e025,14982
spectrax/core/__init__.py,sha256=d1d5e50ca52c0e055f5d5bdff214a11a3d72250de4ca0b676c5dee923ac050f9,1715
spectrax/core/_typing.py,sha256=18c5774ae7b0a61308babf5c1b6b8578149bd346b444273f0fbcd99b960ad17c,4679
spectrax/core/_weakcache.py,sha256=fefb66ebc73f143a8f5cc45837bb426b4601116bf938bf2d2250b0be79480048,2169
spectrax/core/containers.py,sha256=86bacec1e35fd5259e2d9364efbc39f7a3d1418e8abfb8ec1875df3d7c5b8fe9,12080
spectrax/core/context.py,sha256=89a36dbad7becb66c797f9ceb6f9fe561fb92308808b7fdd8d74e9a2cb75104f,6214
spectrax/core/errors.py,sha256=c65491f2af71ae86921c37750340f42c8ff2e04f125c61a4c972bae5a8fc52e9,3535
spectrax/core/graph.py,sha256=e20e0417a47b9f4a692004f0df31c38667a933bece5aea59dcb2226b091b0381,33180
spectrax/core/lazy_init.py,sha256=3bfe9c5c0c2a76cf01fdf6f254a9f7d0ca4ffdca8151aab6a0ca53496a8fdcdf,1965
spectrax/core/module.py,sha256=a7485ca37cf62a73e7fb45e8d908c1e8e46b9f7216150ca517e59a5cb61a163a,37411
spectrax/core/paths.py,sha256=5d0773be85f3e080098ef838ea64ba93409edf357ca1cecba92f58bb469c70f9,4024
spectrax/core/policy.py,sha256=189e55d07707f22b0fb1fd8c17f1b2cbedc575e819ad1cc7e06d540ca895ab01,3777
spectrax/core/registry.py,sha256=16ba49100ddc02c46fb0a7cfd8347a1b02539ae05e563152099598162bc395c8,3948
spectrax/core/selector.py,sha256=0f10d8da5f298ca3f8c96c908249bb228b61a66ec411bb7c768e35f23dc6e4b2,21944
spectrax/core/sharding.py,sha256=511cf50693c905d4c34529f28aa85ed9cea6f5cdb2bdcd388bae93b928c84875,5321
spectrax/core/stage_assignment.py,sha256=b67b431a8e2dc7bbf42954c1641f4058d6c819c28e1f921af2ab63b27fce905b,4063
spectrax/core/state.py,sha256=b5d9de9bb65ed9c20519f094e7fd1ec5aad4d3789535c6fbd53a063a4963163f,13986
spectrax/core/static.py,sha256=cc2afb53ab1a352192931b9b01368acea6926e69d98dc48084b8e28a7b143814,3160
spectrax/core/variable.py,sha256=0cc4368df80f06399ecf7bae5592d55506c725a2f1f4c948d4b024c9b7973000,33227
spectrax/functional/__init__.py,sha256=ff776da663c9a164dc41eb095c14d5f7f3bb06d495778274401e1f015006fb84,1926
spectrax/functional/activation.py,sha256=d5336e26010d94f73357ac6f7e1b345ff30a71837f000267513d4ff8a95db5cd,3732
spectrax/functional/attention.py,sha256=4ca962b20a5a528791ba3b8bbce051ee8333287f10f00ddc36fb80d30a49d4fc,2946
spectrax/functional/conv.py,sha256=767379245ebe016952dcf770bca3a0910179610f2b60b86c90838586d6cfc3b1,3907
spectrax/functional/dropout.py,sha256=399e33d60d8a92d6bc3c7630ea6393a7857755c293a82ade1497dbfd23ae4c5a,1761
spectrax/functional/linear.py,sha256=ad7590868d35982bc75646b79bcbbde82fb18b87a1a3065e181455f5b33c4a98,3157
spectrax/functional/norm.py,sha256=3d71ac81ac605d35ea956184e15230ac4a90907111fc4932ce1decf2869fbd23,2486
spectrax/functional/pool.py,sha256=88dadb74141e9314e7ddc6e43aaca2642ce1c40b9d3ec3acd2f3d76a050fb0ec,3455
spectrax/functional/util.py,sha256=19e683956a4a375c1e426728c270da0991d03e3825ffdb7276b17d6b320455e4,1356
spectrax/hooks/__init__.py,sha256=31b75546c20dd92d6c8428425f0da8a940397029297dc251f791bc0f1ce2747f,1146
spectrax/hooks/forward.py,sha256=df83e1ea178b7b1348a881cb897bd66f1885cf125217a081c0eaad99c3ae67c6,1789
spectrax/hooks/variable.py,sha256=9df61c2a25b64a0c443fff4134898ceb8c116072013813894d58ff9de1bf6f93,2058
spectrax/init/__init__.py,sha256=7c69558b61f2c88f908d3526570a96dd41da399e071aa92ee215c2b1a41183a3,1315
spectrax/init/constant.py,sha256=b00059450e7fc2f809a551f3aedf2e8f3f5ff56a51653d1c0c533be204d47665,1562
spectrax/init/kaiming.py,sha256=2b55765237bcd2e097d2d22531f7a6e2ca1a9855a09466580b19c24b2883bd9a,2569
spectrax/init/normal.py,sha256=6b886284530c40278292f3d3d68dc8a0996b14197103dc96f7fcd873b6dd1189,1680
spectrax/init/orthogonal.py,sha256=c3b80134e2fdd651e11d685e83b292a5c690d2aed671849ded1417a637e1953f,2019
spectrax/init/uniform.py,sha256=435f92237620ab38aeb8abbc29d926a391c2ade1cab7d95eb2fa26b31f5ef079,1149
spectrax/init/xavier.py,sha256=00268ff71235982976e6246bda4704737f38267bdb599bcec1bb6ee9dee4a124,2760
spectrax/inspect/__init__.py,sha256=3dc6e17d39da007a3181b7f4a999ab76be119bbcf0f17ce99a9db2f346ae1815,1072
spectrax/inspect/counting.py,sha256=806dfba1ea3c588c5c23aad9c6c9185ba674f6a1418046c6c41acf60624fb593,1077
spectrax/inspect/display.py,sha256=8c9a23d20331e9320e12fed2ded579529bf4fd23e408458261a715790e4ab8d4,1248
spectrax/inspect/repr.py,sha256=09ac3f6db4373bbd533d1c0cbe53f97eb0443eddfc8fbb6f365e7fa7ebec1607,3290
spectrax/inspect/summary.py,sha256=079854b89cf3e7d2002d9199e47260acc6a466cceec327b6589cc2891c03611b,3088
spectrax/inspect/tabulate.py,sha256=18c5c619abb5ad2f5d92a37d269ae9bdae0d5b2be470499b157b9e3a1dfffea3,4918
spectrax/inspect/tree.py,sha256=1003426b68be95e7a59b94ff38c4cbcc266edd097133378175e427ae7db98539,1645
spectrax/lint.py,sha256=6aaaa1ad5897eeb923744403b6da9371fa3419c312846aed279c53cd6269ded5,2616
spectrax/nn/__init__.py,sha256=ee09a0849f1014b6b316992953b86d2be10f030e5ec7ddd67c1224b9a30301d4,3058
spectrax/nn/activation.py,sha256=f558e5887dc493544bc8d33c9ce5e097756681bb6ffcc228d8e5802868008cf4,2919
spectrax/nn/attention.py,sha256=90f083ec6cc589abb12f44f198149fefb3290c0c24b954ce2864d0f2bc33aa12,9667
spectrax/nn/conv.py,sha256=03d1a35d44b5633d7e256dc07d772a6af5b1d4b54ba1d54c11d35366722f3e98,10366
spectrax/nn/dense.py,sha256=ddf59aade74bc0c918e88fe13447d9b7f6ed8c867904de220edc6688542c5c77,7632
spectrax/nn/dropout.py,sha256=185f34a6850705db8bd1f92617928898e38d86d53b05b767a6218e626d4bf7f2,3513
spectrax/nn/embed.py,sha256=4f5945d37b41ed137cd092fb7087c0fdedca5e265b97af744c432e1c8b53af29,3785
spectrax/nn/fp8.py,sha256=3b4b1e501da6232e0174fb128de370915c3de7ce0b517e517e65f7349e9188f2,29091
spectrax/nn/identity.py,sha256=e0e4afe5f73264eca6628d7975d7fc4b2eabcc744921f62f12f38122058699bb,1144
spectrax/nn/linear.py,sha256=7994637d8983fc774601f6a250fa8b371cb606f830140b475ef470c991275670,7052
spectrax/nn/lora.py,sha256=7a8e8588f60643c7b4b2436180cbf2aff14cd84ecb0e12fb797303259df3ae50,13114
spectrax/nn/mlp.py,sha256=240143a1a827c49624184ba5b37ec79d55800be4c6c068bba813d6fb3ac70606,4040
spectrax/nn/norm.py,sha256=881ec942ec4e565ad9f098a80e1fb9846697886f95bfeb8a6a542e094877db5a,12613
spectrax/nn/pipeline_sequential.py,sha256=950a610127ce1570404593fb48e212c51f4c978d057a3a2835ab67825d1bd29b,5089
spectrax/nn/pool.py,sha256=eb463168a7d8d8437d1aac44967498af2d298b85959443eaea4e96817c51ee5d,5026
spectrax/nn/recurrent.py,sha256=af8e827292f0f4a46d0c9b6103e86b8bda9b74488099635d000380067c7a89fa,19306
spectrax/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
spectrax/rng/__init__.py,sha256=e3a3b5bc6d410ee9dd1336dbd48dda523299a0a1a9a43ec95d55cdb5768e5f9d,1091
spectrax/rng/rngs.py,sha256=ecb2aeb850f188fb49b86d8346fd43be94ef996d2ead266b4c8a10b28429182d,9676
spectrax/rng/seed.py,sha256=fe54db95c2c4d50908d5b155bb987468e51165f07476d13791e44565931113e0,2397
spectrax/runtime/__init__.py,sha256=aad5101dcaef9d128c74443416c5c52fff6cc7c608941fabb31d7abc80158d2e,2036
spectrax/runtime/mpmd/__init__.py,sha256=d6b53cafee9658b51ebe5d4f41b9720dc3ced66a652a204ac1ffa3956a0bffdd,2074
spectrax/runtime/mpmd/compiler.py,sha256=e25f72c07d0feead40c1766b39573755b78b6906c5e31c52064e8118c0860208,12855
spectrax/runtime/mpmd/markers.py,sha256=6c20e525422c47d51f79ebcb4c056cdfe01c8dd2ebd17b23e7bdb51fb8693808,15319
spectrax/runtime/mpmd/per_rank.py,sha256=f88ba3846205968949a3ad378175e7ee7fe842ab09e53ed5af26d77975c6f29f,13253
spectrax/runtime/mpmd/pscan_compiler.py,sha256=a7ec50cc07a9794c5f9241903c43b662288c4a349e3f6d3f08bcaf726b245515,39070
spectrax/runtime/mpmd/runtime.py,sha256=26c14f73eb4671e552f698c4b5060c475d7adde22a2eed376a4826990b2a2662,125597
spectrax/runtime/mpmd/treduce.py,sha256=8295ee6af00f5ce5c5d54686f55d1bbc2920e7e36fd2b5fe18e5b5ddf5a731e6,16712
spectrax/runtime/primitives/__init__.py,sha256=9b0a78f782fa69f1cc2d299dbdbc878d2bd41308cb52171ffde6a891ebd2fef7,808
spectrax/runtime/primitives/boundary.py,sha256=b7c42b2759606c8a5c38688e2a1f86e3d4725980272be2a51633859a67c00559,2950
spectrax/runtime/primitives/split.py,sha256=1b94b684234f4f577868f5224374aeea48f8be3131bc31e9cea61bfa8bdc6676,10695
spectrax/runtime/schedules/__init__.py,sha256=642addec0dec8d12d5b750c5df4bc6b6eb5d32c52abb5cfd76fd9651bb745acd,1371
spectrax/runtime/schedules/base.py,sha256=edbafdd4f6b3250fd59c3f9fd54855201b4e74870a8e78510f9c0805c0f16804,7751
spectrax/runtime/schedules/dualpipe.py,sha256=0d21076e50f24044a5034e785284e7a3d2b67c2ca12af70234becdfb8f1ca9f0,9999
spectrax/runtime/schedules/fusion.py,sha256=0767671317ac7e52f0163caf79582833cc641c50cbda7d62ac4419b41c5c6929,5047
spectrax/runtime/schedules/gpipe.py,sha256=96815fdd830fe97176bbf84160c0a17e7578eb56fd1bcb33982fb8beb907fbc8,2813
spectrax/runtime/schedules/interleaved.py,sha256=69b127dafdf26d892577790b6663c640923a4b31109d2e12753fce7a746fd981,12199
spectrax/runtime/schedules/one_f_one_b.py,sha256=56a9515ed1c77b0502d2a37800a347f005608068a0fdc9e4517cb9b2c9cf8096,7193
spectrax/runtime/schedules/zero_bubble.py,sha256=e9d45d273c78864561a48e87cd3ef160c0ee83c795cdffe891431e54111716a9,4580
spectrax/runtime/spmd/__init__.py,sha256=4de43f0ef15262f61a641142acec436b8543d7b9fd1922ab37a2f0a4e6b110c8,1033
spectrax/runtime/spmd/api.py,sha256=c619a4bd8db397135caf98d0dd85e4e23da6fa6cf15821ae3307b5ebe6c85480,36328
spectrax/runtime/spmd/hybrid.py,sha256=6b1440b3292070bea5ffc2f5752bd7b3879d46a2feec5ad413c9851d9b5822e9,12638
spectrax/runtime/spmd/runtime.py,sha256=d902ae628cae8190f14f3ce8b22ba7b579d1abe47cd33fe17e583749c8bb93ed,13850
spectrax/runtime/spmd/shard_map.py,sha256=a3e5d3fe0cecaba5783f25f88f4fdbd2cc3b22fcb333a856644e56345ef5db0f,22030
spectrax/runtime/types/__init__.py,sha256=a592013a746d09f76d30f59ee36f2c22a6700aebe83df735c84daca92875a46c,956
spectrax/runtime/types/array.py,sha256=a625f166762bcd119017e50f55529c7a57ce8455c1afe557999c9ef8e45e204c,13353
spectrax/runtime/types/mesh.py,sha256=1bc13f44feb161e0b26336383a4e983a12bb9cf684d276ecbcebff1217130f7b,8829
spectrax/runtime/types/stage.py,sha256=869c01123d3e62c5d69bbcbed0643de6d346509eba8b4b7db7a0097b56d2f282,1335
spectrax/sharding/__init__.py,sha256=b2ec24bc59d180b678efcbcd0c03c788fe81c28675b07aedac6a9f239e41db19,1620
spectrax/sharding/logical.py,sha256=b5f22bb6c19d7fc09e52f8bb2210631bb7f8c0f6f5b0587eaaf440810f305260,1850
spectrax/sharding/mesh.py,sha256=46208e3f3a315d688da8d9b2e209e05501154a746eeaebf780c08892a242adad,26585
spectrax/sharding/partition.py,sha256=8e532e4fced16f150c29c6bc5c87e07e2d28e625304debfefd22cad47b1770b9,7075
spectrax/transforms/__init__.py,sha256=9e8745673a50cfafc749efd365c906be262d1f8b37ef1bcf64a18ad0e1122c17,1813
spectrax/transforms/control_flow.py,sha256=a10c1230054fe4310b643195433c506572b6e8e36aa62e9d7efce8e94ce9409a,14503
spectrax/transforms/eval_shape.py,sha256=548daa3851841ca5db63dd5b21aaf11cb989729f8bf8344ae0a31135ebdb76aa,1818
spectrax/transforms/grad.py,sha256=52379aa52a522763ba397e55b8a6c13d3d292159cc906532364ab33b1e4309ad,24596
spectrax/transforms/jit.py,sha256=27b50ef669e0213cc5529108235a32071cdcf7675d967b360fc8542faadd45d2,13862
spectrax/transforms/remat.py,sha256=32268b3205d0f0a6b7988c047290bd7a3d279a47f93304784f24285bfc5fd2dd,9018
spectrax/transforms/rng_axes.py,sha256=868e37e97bc82569b7e7d38330800a946ea1c386ad299814ff5eb5b8ee3abe10,4480
spectrax/transforms/scan.py,sha256=b1cd97971b6c07dbfeab69b28badf18b1ace74ed7233a14bde20173d8f6b4ba5,7959
spectrax/transforms/split_merge.py,sha256=b6d39150143cb1f7bf867c30f42476d385930fc965fa68ab1e301c063f1e56bb,27839
spectrax/transforms/vmap.py,sha256=3bad614456fa312650e252b528fb1a443d75488a79ed473939ce43708c84b644,5426
spectrax/typing.py,sha256=f8bd421699de0ff689ec03ba93b4c35afe5edce7649521d68c30e4eaba4e47b4,1387
spectrax_lib-0.0.1.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
spectrax_lib-0.0.1.dist-info/METADATA,sha256=78e36f28c8c11280fdfe2c7c523131b9d5840edb4f5440525400507a25ecdb5d,16814
spectrax_lib-0.0.1.dist-info/RECORD,,
