spectrux/__init__.py,sha256=c6fac8b5af6612a03746f80cca9c7203b0a03ccc037e05567f1e6519db50d26c,4450
spectrux/_version.py,sha256=7d62bd0504009d0644cbb5a9100c44714865b783712c980643844b312a06854a,761
spectrux/api.py,sha256=62278c0640929b8bad0ac20ea4df9f6e4496001e66813e81aa53596141825997,12481
spectrux/contrib/__init__.py,sha256=e9562327fd05faa9a7fe8bac161069d0a0b6583edd1f6b0c3698af6fd2bd7ff6,884
spectrux/contrib/optimizer.py,sha256=14eef766058d9a640b71e592904bdcccf2dead13ad653b39adad4b771ce37b39,14978
spectrux/core/__init__.py,sha256=5e0ba92f103b97253104884a0ae52fc2d73abdc38f1923828a85228339956045,1715
spectrux/core/_typing.py,sha256=2f50d0cf782ec0dc5fa10768c838466e04d9ff2a4f0374f56101383d50bc875d,4679
spectrux/core/_weakcache.py,sha256=52fe7247c5419298569e6b75f9e992757078fe392af6066111b2a847202037d6,2169
spectrux/core/containers.py,sha256=16713185743487d118f43c2c9caaff481a7edbba38b9e5db7ef335db8b99bc62,12080
spectrux/core/context.py,sha256=0323d06e7e44270429780e33ffe981f239d83b79cbf4a7b95f886f92eb04a8a5,6214
spectrux/core/errors.py,sha256=128c5202137b9ae808a36666c4a35da1db315e033501948960a1f1eab77e5c1a,3535
spectrux/core/graph.py,sha256=b43ec1196da661674550b62846ad2ec8a33c1620232bb693aff568fa7ef9802c,32505
spectrux/core/lazy_init.py,sha256=3bfe9c5c0c2a76cf01fdf6f254a9f7d0ca4ffdca8151aab6a0ca53496a8fdcdf,1965
spectrux/core/module.py,sha256=b589bb89e560f53da3b86fd59ba335e91a5c86b207bada7e73ad54873925bf7d,29845
spectrux/core/paths.py,sha256=7adc6ca4440de944fa9a96bc7f421501c17d63a6f94830ed437d4fae76f73a08,4024
spectrux/core/policy.py,sha256=189e55d07707f22b0fb1fd8c17f1b2cbedc575e819ad1cc7e06d540ca895ab01,3777
spectrux/core/registry.py,sha256=2ab8433ee4e9ff1f638182924b6dadbba573b770da7777e89f83d32f14333267,3948
spectrux/core/selector.py,sha256=d0f6633a30b3d789c47e8a8b65720fdc3c659f7a9a914368780eeef12c4b0a08,21897
spectrux/core/sharding.py,sha256=b184d0d56e9e336ce72dc744cbd2d7c15a4578eff642daee0fc113aa033c2f17,3787
spectrux/core/stage_assignment.py,sha256=814bf15c23bb2cfade236137411ac90259ef0913ba1ae552b410b75d16b30c81,4063
spectrux/core/state.py,sha256=18a11d6cacd2d064bc9deade6efc5b5487f963d032394339f708ae74ba4116d4,10964
spectrux/core/static.py,sha256=2e6368b93bd5cdbc1ec759e73b7a786194d22c0959fb4b96797cb1743d16ec9a,3160
spectrux/core/variable.py,sha256=19cfab1d00794aeb8ca688fdd44b73dcd77aedd96169d6183f21abc7dff971e7,29751
spectrux/functional/__init__.py,sha256=c67ff3fc2d32713607ef89c094938fd8994b2ca41138bc74072a8f26c1f1a54b,1926
spectrux/functional/activation.py,sha256=f09497efe8cf523490c21bcd7fe17b83c9046d8982b932f74d48419d8b923202,3732
spectrux/functional/attention.py,sha256=4ca962b20a5a528791ba3b8bbce051ee8333287f10f00ddc36fb80d30a49d4fc,2946
spectrux/functional/conv.py,sha256=767379245ebe016952dcf770bca3a0910179610f2b60b86c90838586d6cfc3b1,3907
spectrux/functional/dropout.py,sha256=399e33d60d8a92d6bc3c7630ea6393a7857755c293a82ade1497dbfd23ae4c5a,1761
spectrux/functional/linear.py,sha256=ad7590868d35982bc75646b79bcbbde82fb18b87a1a3065e181455f5b33c4a98,3157
spectrux/functional/norm.py,sha256=3d71ac81ac605d35ea956184e15230ac4a90907111fc4932ce1decf2869fbd23,2486
spectrux/functional/pool.py,sha256=88dadb74141e9314e7ddc6e43aaca2642ce1c40b9d3ec3acd2f3d76a050fb0ec,3455
spectrux/functional/util.py,sha256=19e683956a4a375c1e426728c270da0991d03e3825ffdb7276b17d6b320455e4,1356
spectrux/hooks/__init__.py,sha256=a9f4a54613f48285c13fa65bad3ee1ded72c690f07633cba21080e25aa505968,1146
spectrux/hooks/forward.py,sha256=8e05f745b37f1c179c77bf7903e6f91f4cb3f24645d5b207054a9f4aabc7cfc2,1789
spectrux/hooks/variable.py,sha256=dd3d334e8b36daf0b9ddbae14ae49c222dd5cf23e33c4a6cbdd93e9724c68785,2058
spectrux/init/__init__.py,sha256=4d54a2067c715ecf8470ae3e9e6d401d738eedf8103d3a94419a41befe9af09e,1315
spectrux/init/constant.py,sha256=b00059450e7fc2f809a551f3aedf2e8f3f5ff56a51653d1c0c533be204d47665,1562
spectrux/init/kaiming.py,sha256=2b55765237bcd2e097d2d22531f7a6e2ca1a9855a09466580b19c24b2883bd9a,2569
spectrux/init/normal.py,sha256=6b886284530c40278292f3d3d68dc8a0996b14197103dc96f7fcd873b6dd1189,1680
spectrux/init/orthogonal.py,sha256=27297d5a811564707d96e638a1c2a6ee64c0a5cd1104699731501364c8075b21,1819
spectrux/init/uniform.py,sha256=435f92237620ab38aeb8abbc29d926a391c2ade1cab7d95eb2fa26b31f5ef079,1149
spectrux/init/xavier.py,sha256=00268ff71235982976e6246bda4704737f38267bdb599bcec1bb6ee9dee4a124,2760
spectrux/inspect/__init__.py,sha256=3dc6e17d39da007a3181b7f4a999ab76be119bbcf0f17ce99a9db2f346ae1815,1072
spectrux/inspect/counting.py,sha256=806dfba1ea3c588c5c23aad9c6c9185ba674f6a1418046c6c41acf60624fb593,1077
spectrux/inspect/display.py,sha256=29ca31a99623f8c0bcbc07f2bebd46ded096a80c9c08647a844ac8d31fbccd8c,1248
spectrux/inspect/repr.py,sha256=fd6f6c8d26bc3e00c1c73d358ae687ffa644a9d698ae906dba063cc00a79d75e,3290
spectrux/inspect/summary.py,sha256=f5785e0d762e3bf96925b5cf43930c46d9f0922a6afef758b78975643ee8807c,3088
spectrux/inspect/tabulate.py,sha256=18c5c619abb5ad2f5d92a37d269ae9bdae0d5b2be470499b157b9e3a1dfffea3,4918
spectrux/inspect/tree.py,sha256=13dc144b8a09f1e38915bf04c07859642b9d9a8c98fcf669ffa7a93fb05a4aaf,1645
spectrux/lint.py,sha256=176c9f253226b7573f9963ffcf78f63a340059e07126fb8cb438c2c7e5bb7938,2616
spectrux/nn/__init__.py,sha256=50f9f2c30b041e14bb9f519032516e29673f78db28ade0d97f8c8c6c17703f43,3036
spectrux/nn/activation.py,sha256=25a208ba21a8e16cfd8d52d2a7a8162946d3e23b47d158a8952f693ee6b97b5b,2919
spectrux/nn/attention.py,sha256=90f083ec6cc589abb12f44f198149fefb3290c0c24b954ce2864d0f2bc33aa12,9667
spectrux/nn/conv.py,sha256=af425e42fd1386a524961ef81dd1ef1a2e29b31c0ba60f8c472a910bd1d05b43,8674
spectrux/nn/dense.py,sha256=e7b5d3b099dd504ca36376b9dc6f83f97967026aae0048cf06c4d5d6b565ec94,7430
spectrux/nn/dropout.py,sha256=65ccd25fb83fe630e52d4df0e286d5ca5c6336c32f8b997846ebf617b871e16d,2902
spectrux/nn/embed.py,sha256=e7214777d33d4ba255c1bc478ac6221f45ce0e7cda907ae5ac2f4377cd9f6b17,3656
spectrux/nn/fp8.py,sha256=4f0cdf17ee9fe2a5996e1e6da0d073e597e32fd12b258cd3484241ab664e6869,29091
spectrux/nn/identity.py,sha256=9ab06186a6548e3d7cb9772fb96a13ad89dafad924e25f3b550faa8720c34eb9,1144
spectrux/nn/linear.py,sha256=c58033ed895ba67b4f224e547ebcbde1f242adaaf687dedf63f8dc79d1439269,6688
spectrux/nn/lora.py,sha256=e00f93d316bd5bf258f32f750dab6c7e08f4e41196c539c59e442fb053ba0738,13114
spectrux/nn/mlp.py,sha256=240143a1a827c49624184ba5b37ec79d55800be4c6c068bba813d6fb3ac70606,4040
spectrux/nn/norm.py,sha256=fd9f8de815f685c40126f0d5c28c0b2e0ebb27caf48acaac2c641f061aa14fcc,12598
spectrux/nn/pipeline_sequential.py,sha256=393d48063f7adc35211adfdb49b590a8deff4c6ac1999625c0c8f31232355d78,5089
spectrux/nn/pool.py,sha256=02d34b69a05cc4eeccb31ce7f3cbf0a009be0603bee2f908118175b53723d1a8,5026
spectrux/nn/recurrent.py,sha256=f0dc5a701f44bbead8750b583cb39a8c48fd94ded593fda244027a1da320c6ba,19306
spectrux/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
spectrux/rng/__init__.py,sha256=8379c666184c8b273faec114e4691533522e516a04f7cbe74318c555827b18a6,1091
spectrux/rng/rngs.py,sha256=d4f317d6952bcb1dc1b39e4b2c38e1d2e53d20cc182f73e4541a12abbd6beacf,9676
spectrux/rng/seed.py,sha256=a2c67234f8bdf9f269cbc3cfb6d49fd833f3efa1fefe70be8620e6e768b0cc4d,2397
spectrux/runtime/__init__.py,sha256=0e37b478a26273e716f24c036508b40cf1f47795cd05cc348542a3d1385efbcc,2036
spectrux/runtime/mpmd/__init__.py,sha256=d6b53cafee9658b51ebe5d4f41b9720dc3ced66a652a204ac1ffa3956a0bffdd,2074
spectrux/runtime/mpmd/compiler.py,sha256=ff96b80a4b46d534eb14aaa2bc62f918cfdb14f2b803f0c5a6f98a140879cbb7,12855
spectrux/runtime/mpmd/markers.py,sha256=e49e83c059cf35f79c234a8d5a4f79f2c6802782cf1ad67e8b4516ac3e36bc61,15319
spectrux/runtime/mpmd/per_rank.py,sha256=107953d10c469dc8783d3874c5b6b27d9fe8319dd5a9d70dec06a2f2b15b74af,13253
spectrux/runtime/mpmd/pscan_compiler.py,sha256=5b178c2484cec0f86707af5afb7c4659634d024dab603e9407e380dce0d93344,39031
spectrux/runtime/mpmd/runtime.py,sha256=27127d8b830237b9a4270320832f6f1e9d09037f6b3ed6c7482743d0bf225ad6,125558
spectrux/runtime/mpmd/treduce.py,sha256=f170cbecd8011b5a181c00bd553d534ffac3e9d7dfd87bf8aab82442775b2385,16712
spectrux/runtime/primitives/__init__.py,sha256=9b0a78f782fa69f1cc2d299dbdbc878d2bd41308cb52171ffde6a891ebd2fef7,808
spectrux/runtime/primitives/boundary.py,sha256=003228410837b30c76d29ce8ac9f4d42c7d720afe2bad5ed607011b374216230,2950
spectrux/runtime/primitives/split.py,sha256=81263ba295066f63a10f2091a046c4f8a6e1f7ebf9ad99e87885f77400de8d07,10695
spectrux/runtime/schedules/__init__.py,sha256=642addec0dec8d12d5b750c5df4bc6b6eb5d32c52abb5cfd76fd9651bb745acd,1371
spectrux/runtime/schedules/base.py,sha256=edbafdd4f6b3250fd59c3f9fd54855201b4e74870a8e78510f9c0805c0f16804,7751
spectrux/runtime/schedules/dualpipe.py,sha256=802abf1b54ec29e920f4dd2fff5b98529590e9427a35d4b48a24f6cbba0bf0ef,9999
spectrux/runtime/schedules/fusion.py,sha256=0767671317ac7e52f0163caf79582833cc641c50cbda7d62ac4419b41c5c6929,5047
spectrux/runtime/schedules/gpipe.py,sha256=96815fdd830fe97176bbf84160c0a17e7578eb56fd1bcb33982fb8beb907fbc8,2813
spectrux/runtime/schedules/interleaved.py,sha256=69b127dafdf26d892577790b6663c640923a4b31109d2e12753fce7a746fd981,12199
spectrux/runtime/schedules/one_f_one_b.py,sha256=56a9515ed1c77b0502d2a37800a347f005608068a0fdc9e4517cb9b2c9cf8096,7193
spectrux/runtime/schedules/zero_bubble.py,sha256=e9d45d273c78864561a48e87cd3ef160c0ee83c795cdffe891431e54111716a9,4580
spectrux/runtime/spmd/__init__.py,sha256=4de43f0ef15262f61a641142acec436b8543d7b9fd1922ab37a2f0a4e6b110c8,1033
spectrux/runtime/spmd/api.py,sha256=4f6fbf9f6175591d2e396741c4716ba8e4697a13c3d9d4e1d84a8229b9a188bf,36328
spectrux/runtime/spmd/hybrid.py,sha256=6b1440b3292070bea5ffc2f5752bd7b3879d46a2feec5ad413c9851d9b5822e9,12638
spectrux/runtime/spmd/runtime.py,sha256=6c872119c987caf769e6c3d60e01c878e205dc7cf44db164882f48a605e8070c,13850
spectrux/runtime/spmd/shard_map.py,sha256=d0837ba626b6d478b45474534d2ac8e022887ccdfa938860ecf905e6b3871d65,22030
spectrux/runtime/types/__init__.py,sha256=a592013a746d09f76d30f59ee36f2c22a6700aebe83df735c84daca92875a46c,956
spectrux/runtime/types/array.py,sha256=aa4460949283e60cacf5c9eb5f7079ff5b3c82f5a347bab0709903abbff3f91d,13353
spectrux/runtime/types/mesh.py,sha256=620ad9b3631bf4a1dcaf80edd97c4ecda6ee4e4c40a61372415c952da6cd50f6,8829
spectrux/runtime/types/stage.py,sha256=869c01123d3e62c5d69bbcbed0643de6d346509eba8b4b7db7a0097b56d2f282,1335
spectrux/sharding/__init__.py,sha256=fbb28b095487097b23f7f271b74b46bf755d862738ff0d09962855da4b3c5c95,1620
spectrux/sharding/logical.py,sha256=b5f22bb6c19d7fc09e52f8bb2210631bb7f8c0f6f5b0587eaaf440810f305260,1850
spectrux/sharding/mesh.py,sha256=aced0f5088e6884ff8575b39bc8f8cd1f44cd195ab5255e317ba34e02826788c,26585
spectrux/sharding/partition.py,sha256=0349c4872a938bbaa7e3afe1f68134ab68e4e8a4e623e6314d81217166a0cffd,6846
spectrux/transforms/__init__.py,sha256=8c68334c9d95e99333fde1c2e36c632e539be5d60602b79df381bb5fcb9223bf,1532
spectrux/transforms/control_flow.py,sha256=2aaa63ce1a35a72af1fb047b2ff41e305ca2acb9fcd2a220d38e1f7cc0753639,11308
spectrux/transforms/grad.py,sha256=cd69a7676039fb938658efd5709bc64be2a11a6908036e6925c567649e7b7b0d,6249
spectrux/transforms/jit.py,sha256=b81164e65e21ee7574f13a47ef8ef7bfc62b2308ce6193f1e497b7b4b99458fc,9847
spectrux/transforms/remat.py,sha256=e61eb44beb7ab9c9c8fc6c544e29b2c3f04eb39258b2a0b3757f9102105a2aad,5860
spectrux/transforms/rng_axes.py,sha256=cd5f91ab98b70df6e40679f227e0f8dd7a20bea4489b04a19ea3d9c841e3bf8d,4480
spectrux/transforms/scan.py,sha256=06a3bc12b443d81ffbafa64a9d84a68f6378b5ab4882a28418a7bb0c6fecc5c7,4074
spectrux/transforms/split_merge.py,sha256=c420a0385dd036fb09ee82a0e54983c9a10b34b9bc56461712ea0de8bd44f546,19758
spectrux/transforms/vmap.py,sha256=7e66a549d4f4517dd3585474bf9922a59212e4267772c16f24898d1ed2cbca22,3198
spectrux/typing.py,sha256=8c08e212bdbdbc47ab390cc7813d51a6aec21788f9473ed50cf2249932d31fbb,1387
spectrux-0.0.1.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
spectrux-0.0.1.dist-info/METADATA,sha256=6b9ca28d1a97a74265b30b9944b30124ab2e0908f3cf308f5877b4deb0f2d815,15902
spectrux-0.0.1.dist-info/RECORD,,
