spectrax/__init__.py,sha256=19f481575096a607bb969be0eabf5ca5e4c841d17e35af27f8815fc496ceb449,7776
spectrax/_version.py,sha256=ae86070acad22eb3d35f7d644cf1cd0a5a1fa1113767ab30bbe934038b0d1bde,298
spectrax/api.py,sha256=a3c6b3c464175bec01dfeb7fb20c52956ea61f8bce770e1b5211d138d47d1fe0,18820
spectrax/common_types.py,sha256=9e7351db2b0c69a300bb5522bf25be28c7fb0f65738b49399a01905c36ed711f,13416
spectrax/contrib/__init__.py,sha256=ea0065ed5f29814ee7061c102ca812103083c194c9a39e1b77944baf03ec0ac4,757
spectrax/contrib/optimizer.py,sha256=85bb5021c5d80d95de8b84697501c56298424dbc34941b2987f0a22399171167,17419
spectrax/core/__init__.py,sha256=443f2374f444df7d61fac8385df74ab15ecb732a38827cf83f4f6d6f5b1891ad,1352
spectrax/core/_typing.py,sha256=552e7cced334d42c6041321df063cb8bf6aa9a93968a4a4863885b26e93e1c6d,4624
spectrax/core/_weakcache.py,sha256=52b4eb5ad497b4b475b86142548c5b2be6b9118bb6739e45fae0920bc9d14eca,1706
spectrax/core/containers.py,sha256=93d008ec7c51275149735f520e7087bb4fae4cbffe550ace096a98414337a5b7,56103
spectrax/core/context.py,sha256=2af3c02a6a4f7811053620e94d35a743856d2187eee5ca7ddb159b22f289385f,5751
spectrax/core/errors.py,sha256=0b51f86c6f7a377535ac3149b86e84bc334986fa36a5c8e09881719e212d9667,3072
spectrax/core/graph.py,sha256=7bf321df3f6dc673f92dcd1c6437723fc784775bc5be3b5a2aecfe2c02af88e9,36195
spectrax/core/lazy_init.py,sha256=43f80a08df8d3ed894ea081f4564c48959534bd0db6e09153d26c3cb40e09d25,1502
spectrax/core/module.py,sha256=8e8392dca84469765b6a3c6ee820c9eac93ec5015eade38fcb2c26cae3af0e83,53892
spectrax/core/paths.py,sha256=cdfda958c1136aef7e342bf1f9332887738a0659a8bfb8a0ba81b345d5c8c348,3561
spectrax/core/policy.py,sha256=47ef4865cdaea89395fd2036d07ca71b3a659451ba6d85afee19ad3901ceef43,3314
spectrax/core/registry.py,sha256=1c9cd2a0d69241f5e66e9b0175f780ee26bbfa11fb2eb3c601ab9c3002c00bc9,3514
spectrax/core/selector.py,sha256=0b1cb0998a01c906b0b44eb38b30aae254b7e2e6e7e49e5311f247f70aec68a7,23872
spectrax/core/sharding.py,sha256=e02eeef0ef0a259d6822a155306158067a8fa681628b5a7e140b978e19d0e9b4,6384
spectrax/core/stage_assignment.py,sha256=5c64ad2d749463faa9ce6d47072a87d29456a03398f5188c5bc2372c0b5fd568,8036
spectrax/core/state.py,sha256=8f7a4e15df50bbab9764dca3b70d2d88fdd6b8e652e9523fe3052e029e9e9bb5,26581
spectrax/core/static.py,sha256=0afe31c73476551e679e97f2685819b35ecf41352fe669661241579e89968241,2697
spectrax/core/variable.py,sha256=c2f4af18b3b39661fb9665c47aaec05f8003cac31450e66f825541e89434a76a,43562
spectrax/functional/__init__.py,sha256=9dd61b9f0110ff526152cc717d0b5b99440881e58f23d040e99dcd37a6a81c80,2297
spectrax/functional/activation.py,sha256=7f9b0a06ef3045af2648005bc0d0000663e02593fc0d33406e83a37b7b1fbf08,8432
spectrax/functional/attention.py,sha256=413237fa41ebfef277ed9ce4e87093806f2048bcb0bb775767cacba34dce0b63,3036
spectrax/functional/conv.py,sha256=44287d1a9f6376bb1f86cfe4c42376a8e5ac17d424e784d8e981e85317462801,4994
spectrax/functional/dropout.py,sha256=ff9bb906cd7b97998c29cb49f287fa3bb906f809aaa08fc37efe9c4583c17aa5,1728
spectrax/functional/linear.py,sha256=6975199d746423cc2cc11ba41b550b6162f6798d978c1c64bcc210b3e8b586f6,3122
spectrax/functional/norm.py,sha256=368500f7eb952a522774e0024de46caba7ef53c17152c51607f6da5c721c64d5,3545
spectrax/functional/pool.py,sha256=393d6388e95db7816b6b48f12a71ce03300f88f4165d87897ebaaf23e79e2019,6174
spectrax/functional/util.py,sha256=c3c91bd2a85d7ded9b4d10c0824e0c0f117820e14d2b8ccf6088100e0fd59fde,1480
spectrax/hooks/__init__.py,sha256=7a95296fbc1bb94f3fba7544d01bfb549ab077d5e3ad049c99c69a1dc3377090,1128
spectrax/hooks/forward.py,sha256=17ccf7e3c40590f2bfaf91b521d2f94665a87c77faf1ed88dbd97b82072938c8,2135
spectrax/hooks/variable.py,sha256=d291c09f0bd4a2c4aa2d841574521dec5008955c912047454dbf8ce78e3c7b99,2662
spectrax/init/__init__.py,sha256=4400c82a2c82d28c5d7a0ac669aded3aea8a170b0991cc44b25e29c8fca55353,1483
spectrax/init/constant.py,sha256=10926cc932c2e63beb452dd42b861169c663648e53be3024e1ebe97e966ddb0a,2238
spectrax/init/kaiming.py,sha256=4fc5d2049d25271ac61c4ab1a9f6e9d2fe89e6b87d31de22439fe7206d4caa99,4021
spectrax/init/normal.py,sha256=c9da96c74a983d6741f5c9513d9cea3de200c34bad816886a9ea32b6d299c5ea,2527
spectrax/init/orthogonal.py,sha256=b642ac93ebde66fe18533b4c1522581fe1326cf82e65d65f1d61c7b624d28637,2621
spectrax/init/uniform.py,sha256=5338911602418a5781c5d342a6bdb713a9e703a71ce15b67b2fb6582157cde2e,1174
spectrax/init/xavier.py,sha256=e3a0a01f6da7fa3554c1a4dc53f43c2bd66e130aac4355e74597d4b44360b2ab,3632
spectrax/inspect/__init__.py,sha256=924547e640709ee38bbee61c15c2475029b93af5dc18c6260c7a07855538b157,1304
spectrax/inspect/counting.py,sha256=aca0003fb388bc828d00a3f38080ac6ca49acfe0bf8bbb12fe3833095d67d061,1149
spectrax/inspect/display.py,sha256=79a6c15e04a3f16e3414094fb989349f2f787993e162f4d097b7998ed2b610d7,1512
spectrax/inspect/repr.py,sha256=9f995c963757ae2b5d8bc00ebae4c3221db34e5e6c100176e66acdba766e757a,5267
spectrax/inspect/summary.py,sha256=45e217efb2967eac1f07e3d9284d248e8013b8edc6e47011c38c74e41f7a7de2,3789
spectrax/inspect/tabulate.py,sha256=e1cd36bc0554ca387c3a79af3fd0fe18c6c2c70e482f0cedf3d0e0273796102b,8438
spectrax/inspect/tree.py,sha256=453ab349f18d9b34c3dd5dcae0a484e6c2d762f8e570c16818c80f1fbc055c2c,1969
spectrax/lint.py,sha256=11023ff9768dd092fd2be42135d6d5200905b43575144fb7714e92bf6c18b6df,3133
spectrax/nn/__init__.py,sha256=9329c3964c7a1875efddcaf9d91278c35c5ad86fa8fe63f7a459e4e683792e22,4016
spectrax/nn/activation.py,sha256=eac8aa25b45c9594f940f933ecfbcccb873fcc77437e13b238da8e4996e133aa,5757
spectrax/nn/attention.py,sha256=1e4f6e7fb5fc6443e43465b41ce7ebd072893ba0341c9f89e4bc47fcec330e1a,15735
spectrax/nn/conv.py,sha256=af471781e00b819e913c881290a357ebcb70f0807c05e4d2b23b68332f595c55,16958
spectrax/nn/dense.py,sha256=c6d7611905133f59006b6fa47ec8719828ab826d61729192de21ed70224eef1b,11703
spectrax/nn/dropout.py,sha256=dad305854aedf1f11c1e595c0916cd4737b46355df2a219f3ab734a5c472db58,5472
spectrax/nn/embed.py,sha256=6f209e872a5839b91a4a78d72c444c77cc25c82d126197b7a16ec86c86455a1d,7362
spectrax/nn/fp8.py,sha256=091a4f53f5a3c882ac09240bb6476dbb6c0bb8034351759e1e10153a6aa8562b,34194
spectrax/nn/identity.py,sha256=2473d3833a6db02a62b394ea1f1a7273f262c6b319c1fc9250ed07ee9fd90f70,1698
spectrax/nn/linear.py,sha256=41de95d45e54bc73b50baf6b995053920093541261a89524a1b251bb4ececde3,11040
spectrax/nn/lora.py,sha256=95d9dc15935245b3817dfd24ea761f34252cc3ca6997c41a3c0ee59340e70f99,15031
spectrax/nn/mlp.py,sha256=f5e36ec725b1ae208480e91ca3afc54379d2938b63ad377eea910d51731888ed,5253
spectrax/nn/norm.py,sha256=1bb22d02ce009ab0b0b50bc8fe1859b6f92f5fdd33501ce0d4a5c6b63a5fbc03,18933
spectrax/nn/pipeline_sequential.py,sha256=64bb8818266923f1a29413a38d9b887b98fe78e655973d1f5406dae242de154d,6158
spectrax/nn/pool.py,sha256=1cab8e28fa11cb71b013eaac692fadd7a4d62795c3c345b122b7187a2124a6a0,9458
spectrax/nn/recurrent.py,sha256=e64eb0835e01f5ef6b33e33979230fd7e6f9e8fdf86b62e38608470c6c08f518,33697
spectrax/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
spectrax/rng/__init__.py,sha256=5c721a22a1b3dafa126991902230312321d51a536e16891c7b7930eb4b29e627,1422
spectrax/rng/rngs.py,sha256=5b853a04951c4146bd183593370c644526e69568432340f06b39c5e3d8578e16,20002
spectrax/rng/seed.py,sha256=a82a825acadba997a9f29e152f1a313f04ce9581ade5f5a1ad40894d978335c3,3080
spectrax/runtime/__init__.py,sha256=f0ed2901558641815269c0e487dd73ad84910b36e9ad5b7558b9357fd28c71fb,2696
spectrax/runtime/mpmd/__init__.py,sha256=98f5b1b223d92647b0dfbca649cf613e1c204e7bff9f5b434dafcb1025066ab3,1611
spectrax/runtime/mpmd/compiler.py,sha256=4dc22bdb48d32d3a7f54107cd1ca4632f468e0d627ec381a68fca4808f475a4c,15886
spectrax/runtime/mpmd/markers.py,sha256=fc59961bf6f879b8a6c2ed79c493889f3b81d5c423dc484b5e2b8345281e2f6a,24667
spectrax/runtime/mpmd/per_rank.py,sha256=4b7800ba7c442263261def64e6eb5db4b423d849cf5b8a997097a0b1837197a4,13992
spectrax/runtime/mpmd/pscan_compiler.py,sha256=070ca7b43729d09dd3d39b889d5d07a86e29f9052c00a37f68252806ef25659a,51596
spectrax/runtime/mpmd/runtime.py,sha256=1e3ce19b280ac0ffdff2eaef6234d55d9a598700149d9d3b333ce59cd54eb0f4,256780
spectrax/runtime/mpmd/treduce.py,sha256=d318580817d0e5fc957414f3578848bf4c55588ef27f21da1a847597907e1845,20566
spectrax/runtime/primitives/__init__.py,sha256=0978c4c75a1f30dee61116a8dc6a8e903f0c9b1be88c4d4a177605f2a30dbd7a,1059
spectrax/runtime/primitives/boundary.py,sha256=3ff3dad27f98d59d92a9f7124579da13ddde5d365a3b84f03c631105bf00bdb5,3295
spectrax/runtime/primitives/split.py,sha256=39acbeeeee2c4df5aa7392642cfddb64a9d81c222a5663fa8f701f20040797b9,19576
spectrax/runtime/schedules/__init__.py,sha256=a4de7bf63f82471aa41d7d713db39046449d186712108b29e1c7e1c3a2ae72f6,2666
spectrax/runtime/schedules/base.py,sha256=04cfb2ba87db640b3009d5c064e37fb56ef3bc7aa8c2a109bafe49dffe2cc17e,11228
spectrax/runtime/schedules/dualpipe.py,sha256=bb83977ae8e73859a1fe3ae3f43840e14b33b3455fff41e83fe706eca5210524,14702
spectrax/runtime/schedules/fusion.py,sha256=76abe8ac813667f2524ea3afd033bd405bc65d21a01483c1ca806e3fab8259f1,4584
spectrax/runtime/schedules/gpipe.py,sha256=a58b7203833679a5d9761845b2e745481c7536e5e9575bd23ab2815c8c98cfea,3075
spectrax/runtime/schedules/interleaved.py,sha256=0fe0c4ee9c48d5f5ff419795e55792577685db63ac511cce50e4e1806ee5f200,28201
spectrax/runtime/schedules/one_f_one_b.py,sha256=5bf1e7f3b8e2957af7e2f83a1dacc8f48af5f72936b3d35702d0ca6fb807861b,9448
spectrax/runtime/schedules/zero_bubble.py,sha256=45ad70e51198690f8940845611787c63b984e231073e37e85116dc8b1d1be163,4852
spectrax/runtime/spmd/__init__.py,sha256=bd20fd6739a287b8dd97dfee3c990812e9a0171fff9e997fd1f23bf8be12e934,1942
spectrax/runtime/spmd/api.py,sha256=6b742743dd5516c44108d38aea8b88c003d6306c762c5693ee29f6109415a569,46623
spectrax/runtime/spmd/hybrid.py,sha256=ab871bbed5b80262c02a0bd631da611329c5773a96aa247a13dfe5bf58a0f004,20466
spectrax/runtime/spmd/runtime.py,sha256=a8eaeb56a1b7088a58039052a18b4cc34d4fe4a6a78f68f88174035c9f0af6d7,19418
spectrax/runtime/spmd/shard_map.py,sha256=23c9a1ba4dda60765c083ffc5b64753404cfcdd2f673bcbfd4123ff705d14fef,33210
spectrax/runtime/types/__init__.py,sha256=289e22e11c5e6ca492547ec7416d8143f8c6bed5357eae86f58dc44154b9846c,1377
spectrax/runtime/types/array.py,sha256=f482ccd3af4ead44f1c2ab5df34ccf328811362be86fa045e3ddaa5598196dc6,12890
spectrax/runtime/types/mesh.py,sha256=5ff556aa1b1c526aa89396f2acbd8450afdeaf7fb07be4b8fc127e5a5db5b372,8364
spectrax/runtime/types/stage.py,sha256=ccd2c81def079a7d210003a9da7bdb6ede3b923a32b41b208f3ef2786170006f,3479
spectrax/sharding/__init__.py,sha256=f85bbeea9ac38bcf50d448145e889f26f36da392fa2b5cb332e2f5c572f90780,2498
spectrax/sharding/logical.py,sha256=49e9e68938191c14f2bf56da33f9c3867170b9e8812a64dcc48d1744dc89e736,3198
spectrax/sharding/manager.py,sha256=2651f4c4628a7b5754a594fdac9d089bebbef1bb05459f47ae0a698414f2c4f3,40570
spectrax/sharding/mesh.py,sha256=d7b9ac788bf94105cf3fa418ebbb68f4268885f933a536d7c19e9b7b4d9878fc,34299
spectrax/sharding/partition.py,sha256=a089e744ffbadf24c1f763d3a535845aa58487f2e05d650e8151e1aae8b38f5f,50878
spectrax/transforms/__init__.py,sha256=ce80363cf9d80f96f22d463c2139090adfd274eb99282731dd3aae79772f01b0,2007
spectrax/transforms/control_flow.py,sha256=d9260eca8ea8672e16a2073780e704e7f994e7e129e5631485735c4f172f1ee1,27033
spectrax/transforms/eval_shape.py,sha256=15c1fcdb018511c67580b80f258af9ac357ceff14112a633b37126898acd0d95,2716
spectrax/transforms/grad.py,sha256=6cc0892a3ad98fa2c6bbca5679421155510b59cf9fe5e818101a6e9c9258bec6,43760
spectrax/transforms/jit.py,sha256=46e3df24aae48dbc38cf55ddde2e5156fbacbe24527ff552fc46e8479cde53ee,22651
spectrax/transforms/remat.py,sha256=c542b5662c1034c95418280c02d70ee6c05f59b04e5c9f7cc32440e457d475bf,13522
spectrax/transforms/rng_axes.py,sha256=8d2988cadd3b7066718c1593cc1a1303ec17c6a06a271f89e413ab9b4c70a19f,8309
spectrax/transforms/scan.py,sha256=be8eb4265f58ad0122676f3931d8d33175cd1879161b780a2652722ebc8b0833,9676
spectrax/transforms/split_merge.py,sha256=a64ae51fa33cb2933a87c9a3067acfdc74164726f0ae80cd88997d489d542daf,37550
spectrax/transforms/vmap.py,sha256=597d09d29e9d1757424cebae952259452787be6c490ca47b13f1945138bf47eb,8746
spectrax/typing.py,sha256=e5a4e56cbd780e4c7069beb4ca60585e34f5ce10976f1edbad56b74e11f679bc,924
spectrax_lib-0.0.3.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
spectrax_lib-0.0.3.dist-info/METADATA,sha256=3789dd4ad51e38eb8953df15c04d6935943f6927eba34a130b2506cbde018804,28282
spectrax_lib-0.0.3.dist-info/RECORD,,
