spectrax/__init__.py,sha256=317a5e2dd1af111b4b6458350f0428aef49626e2a3d3f91909e38fb85c79b5cd,4777
spectrax/_version.py,sha256=eac78a9d5c6a3e065ee7cb5b58b51d29abf5375530e08bd695a3192f67a5c547,761
spectrax/api.py,sha256=cf39d94e95cabdf55c0a2098455211a313bd5ae2c3d36f8e0591aa6915bf273e,12337
spectrax/contrib/__init__.py,sha256=0f5b6523674dca96558d17259f61e920c193cf21e4ffa31da11055e7e825badd,888
spectrax/contrib/optimizer.py,sha256=70e9bf0358497e99bb96dd2489ad409e7661a11cac4805b2daa0f617d842505a,15194
spectrax/core/__init__.py,sha256=c669c58f65fcf0902edf04a98cbb45332e42e692133050dcf727fdad4cd2736e,1815
spectrax/core/_typing.py,sha256=18c5774ae7b0a61308babf5c1b6b8578149bd346b444273f0fbcd99b960ad17c,4679
spectrax/core/_weakcache.py,sha256=fefb66ebc73f143a8f5cc45837bb426b4601116bf938bf2d2250b0be79480048,2169
spectrax/core/containers.py,sha256=fc374958198d52bdcf445e2ec8aab8ba7120834b588150353e1b0d8e1e238ed4,46618
spectrax/core/context.py,sha256=89a36dbad7becb66c797f9ceb6f9fe561fb92308808b7fdd8d74e9a2cb75104f,6214
spectrax/core/errors.py,sha256=c65491f2af71ae86921c37750340f42c8ff2e04f125c61a4c972bae5a8fc52e9,3535
spectrax/core/graph.py,sha256=0a5df7b2e8456fcc258e44c657870d6895ae3a19bff8fd2c66c742bdf07fe7b7,33627
spectrax/core/lazy_init.py,sha256=3bfe9c5c0c2a76cf01fdf6f254a9f7d0ca4ffdca8151aab6a0ca53496a8fdcdf,1965
spectrax/core/module.py,sha256=f70495fa0637c28a8f65d3e830531985fd932de0ff55a89c626177e26c9a61eb,43167
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=2c6b9239b75dba4f566569df04d8ccad901bcb7c2923d3ba83877249a8530284,22603
spectrax/core/sharding.py,sha256=511cf50693c905d4c34529f28aa85ed9cea6f5cdb2bdcd388bae93b928c84875,5321
spectrax/core/stage_assignment.py,sha256=b67b431a8e2dc7bbf42954c1641f4058d6c819c28e1f921af2ab63b27fce905b,4063
spectrax/core/state.py,sha256=a3bcbd37f3bf4de039346edc3dc22fc0cc489065a453edc95d99ccb1b3b9edc6,24733
spectrax/core/static.py,sha256=cc2afb53ab1a352192931b9b01368acea6926e69d98dc48084b8e28a7b143814,3160
spectrax/core/variable.py,sha256=0e734dc42706637754eae23a5f01a4a0758162d118597465e61b4f66488593b3,35044
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=5c7ef5e23d47084f24d60893026d7304d3b70db79492bdc452e43673bcc90fb0,1088
spectrax/inspect/counting.py,sha256=92be81d9d5b10b2e10d9678466cee20aace1694cded102062b24651cb6c5dfaf,1093
spectrax/inspect/display.py,sha256=8c9a23d20331e9320e12fed2ded579529bf4fd23e408458261a715790e4ab8d4,1248
spectrax/inspect/repr.py,sha256=09ac3f6db4373bbd533d1c0cbe53f97eb0443eddfc8fbb6f365e7fa7ebec1607,3290
spectrax/inspect/summary.py,sha256=cd483660b1a353b85a227607edb9379acdefeefdcfdf554704034614a3f73721,3108
spectrax/inspect/tabulate.py,sha256=418e39e38db9f2212eb4f6326dc41c4d18d816442302d1548c98e58e161e3545,4978
spectrax/inspect/tree.py,sha256=1003426b68be95e7a59b94ff38c4cbcc266edd097133378175e427ae7db98539,1645
spectrax/lint.py,sha256=6aaaa1ad5897eeb923744403b6da9371fa3419c312846aed279c53cd6269ded5,2616
spectrax/nn/__init__.py,sha256=952a49def982e66f683b544dda42bbe071b6cefed0d3928826b3ecbdaf2885c5,3143
spectrax/nn/activation.py,sha256=f558e5887dc493544bc8d33c9ce5e097756681bb6ffcc228d8e5802868008cf4,2919
spectrax/nn/attention.py,sha256=90f083ec6cc589abb12f44f198149fefb3290c0c24b954ce2864d0f2bc33aa12,9667
spectrax/nn/conv.py,sha256=43b6986563360cbb9a1c982defdb3f44b4dc7b111859c398da69cf8be74a8ab1,10390
spectrax/nn/dense.py,sha256=ec7c7fda071d33a2bbc20925277db6b9cbfd4626bb42e106ac8db1580acf0ed7,7648
spectrax/nn/dropout.py,sha256=185f34a6850705db8bd1f92617928898e38d86d53b05b767a6218e626d4bf7f2,3513
spectrax/nn/embed.py,sha256=f082e31331430199a1e640227a91392be7ae87676b18936be50e775a9459c140,3793
spectrax/nn/fp8.py,sha256=1bc0cf908c92f6a52c0821c9dd8e59075717329fbdc076b7c3e628dd19d653f2,29115
spectrax/nn/identity.py,sha256=e0e4afe5f73264eca6628d7975d7fc4b2eabcc744921f62f12f38122058699bb,1144
spectrax/nn/linear.py,sha256=ace4b4ed112fa4932798460df8a9a31ae59afaf21f36c792fd7c3ff068582a8e,7068
spectrax/nn/lora.py,sha256=20a484fef84f260e0472e04e1c748d7506282d42bc413b9b9dacf9c5054794b2,13134
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=33b4c1527ee60c4b4fa97199c0cc861fbcee2154be1398b2b36b60d8cce66fb3,19380
spectrax/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
spectrax/rng/__init__.py,sha256=e3a3b5bc6d410ee9dd1336dbd48dda523299a0a1a9a43ec95d55cdb5768e5f9d,1091
spectrax/rng/rngs.py,sha256=afc4944f7f3937b54548d2f6d58bdfd9c2bdc948c1c088f000943e2052c8480f,9745
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=9f5b0ca44bce766a99ec0adb2fcc4ec03f75d41d3c815073fc1bd498ad6f941e,18786
spectrax/runtime/mpmd/per_rank.py,sha256=f88ba3846205968949a3ad378175e7ee7fe842ab09e53ed5af26d77975c6f29f,13253
spectrax/runtime/mpmd/pscan_compiler.py,sha256=31c5e661e26a6928380341d52a5e57d6b12462a19a155cdfa1c3d31e3de4375e,41977
spectrax/runtime/mpmd/runtime.py,sha256=4bd578f303e6b2a49c0c5d6af9e9de136009e3ed640578bd6002dacbf3c9d98b,193719
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=5d1d3ccc5ef6f102e4de13f596e3e76358306276fa4577c8beb8c8d5d4ca0b92,12598
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=6b377715187f5c300f5118fde297f469c502fdc9c5359a467da52006b75a9e46,20586
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=c74d88a20c3d1f2482bb79f7cd4ceae93792a86e2ce1556ca481d6e51b2a5aa0,36332
spectrax/runtime/spmd/hybrid.py,sha256=6f94284701199bf962cd1222e50cc51b79112996a3c5d2fb1a4889d6921fabe5,12666
spectrax/runtime/spmd/runtime.py,sha256=a00f49b091dcd10c74cdca8fd88847f3307b6fac86cb792298ce27513c9c1c35,13847
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=46603a2357f653491b442fd5d0519c470cab17eb3109903430e03e3975d5a364,1343
spectrax/sharding/__init__.py,sha256=b2ec24bc59d180b678efcbcd0c03c788fe81c28675b07aedac6a9f239e41db19,1620
spectrax/sharding/logical.py,sha256=b5f22bb6c19d7fc09e52f8bb2210631bb7f8c0f6f5b0587eaaf440810f305260,1850
spectrax/sharding/mesh.py,sha256=f5655da7ca3248b43bf4be6e2e0c2cd6d97b6f07ce49ff89ffb102883f66c2fa,29689
spectrax/sharding/partition.py,sha256=8e532e4fced16f150c29c6bc5c87e07e2d28e625304debfefd22cad47b1770b9,7075
spectrax/transforms/__init__.py,sha256=9e8745673a50cfafc749efd365c906be262d1f8b37ef1bcf64a18ad0e1122c17,1813
spectrax/transforms/control_flow.py,sha256=0dc124a5d3c98663c75d22857246a4a662cd6d9f3ca46b4838e21f9c58aa149d,14508
spectrax/transforms/eval_shape.py,sha256=548daa3851841ca5db63dd5b21aaf11cb989729f8bf8344ae0a31135ebdb76aa,1818
spectrax/transforms/grad.py,sha256=6e17645cffed839fe4c2e34c3bf5e9da9dd4c6d96e64354bbe56c31abd5f1ef3,24614
spectrax/transforms/jit.py,sha256=972872d3aee432037406d86d6b41d05cb9261aa65f11e59c1308ba24d2a1bb22,20450
spectrax/transforms/remat.py,sha256=9224bf354faa4ea7e2eba0659982b5102c80e85bf286942faebd6c3efb79a435,9239
spectrax/transforms/rng_axes.py,sha256=868e37e97bc82569b7e7d38330800a946ea1c386ad299814ff5eb5b8ee3abe10,4480
spectrax/transforms/scan.py,sha256=b8d4049c45366240084711a437dd84f8392250683fef031780cea6ab7c2cec1b,7832
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.2.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
spectrax_lib-0.0.2.dist-info/METADATA,sha256=de3e7d1111cc71723f87f7aad3f3dfc07ffe9346b19906bba291ca42f590c156,16905
spectrax_lib-0.0.2.dist-info/RECORD,,
