spectrax/__init__.py,sha256=f8f6ad6c9ac8c9e04cd547dcda2f40a8192d83462cadd31d5c5e50ab80cf4b0f,8359
spectrax/_internal/__init__.py,sha256=9d232efd43cc5ef5b0e521f6ee4dbb270799dd83ca68d4da2ba267a070a561aa,493
spectrax/_internal/dtype.py,sha256=b1752c73adf65879b89360d2860af581a5ff14816553d1aa56e5c8313153ecc7,2452
spectrax/_internal/logging.py,sha256=77c583b17d8f8c576dddead7333eec0b8f357735dc79617fc25bae9b35b907d6,10195
spectrax/_version.py,sha256=a4d77530063a05cbc4ffec13d4d8d27bb5d5eb1059209c0fb68ac06916d2803b,298
spectrax/api.py,sha256=a3c6b3c464175bec01dfeb7fb20c52956ea61f8bce770e1b5211d138d47d1fe0,18820
spectrax/common_types.py,sha256=bb469a57a4d115dbce1acaf808c4addcc77aa409c7f788971a2de6e3f8bb6399,13415
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=c40c4c0b996d32612193b4c890f4ae9e06217fc65d0895b6b8486ceb75543a51,55884
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=ca6612cb35e8ff6b9045e2682a6b01262572f29f2041538d5c1da8e7ed81177a,26809
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/loggers/__init__.py,sha256=5f99931180f45edd237970e0b2384c3d00564034d6de6d8a2f3de127ef6dd2ab,1315
spectrax/loggers/base.py,sha256=744571b60a71d38aac261f74b9f1848c2b84e852cf7aa98725a7ff95b5051699,9604
spectrax/loggers/console.py,sha256=8ae2ab7010230fb62377331b574744c522c54cf05df3c2118bf7cf48de4d327a,3273
spectrax/loggers/tensorboard.py,sha256=69a913cd3c7d6aab7513a1d05eb3994ff25504de3e819e9ea1025a24d1ee9e85,13936
spectrax/loggers/wandb.py,sha256=030beb6e60cba0b99662410cab18ad8df9d599fbdb6c0484bf9d50f6df89fea0,2624
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=999450a8c9ecaecfabe63f2248688a23ed9292a8c5cfa4fe10886d964b0de6ce,15884
spectrax/runtime/mpmd/markers.py,sha256=fc59961bf6f879b8a6c2ed79c493889f3b81d5c423dc484b5e2b8345281e2f6a,24667
spectrax/runtime/mpmd/per_rank.py,sha256=d8685a7120861ae9a26878ce2fb7e1f66e55f004c6728cc66bfc75e383f6366f,13991
spectrax/runtime/mpmd/pscan_compiler.py,sha256=b7fb8bc4c6eb1cdb339177459a68ba1d9b896d4baab4820d1497e6fc30ba2e32,52315
spectrax/runtime/mpmd/runtime.py,sha256=c91fa90f906e94528f8ed6708df9da9c9ab7f50dfb5cc68d7e8abf66c26de1d6,270354
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=87ecc04cf2e552bfb7a2c34f75c3936862c0094dc31f5732c79c0578a7e99897,33208
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/serialization/__init__.py,sha256=39096083a8acf361cf4475497f5fa8cda322a44215a02bb8e2ab6ac78cc394b0,977
spectrax/serialization/_compat.py,sha256=47d5350535afee096592d3eb8cfaaae8871fdf1dcb58dc6cec012a9aa8b0b1c1,2467
spectrax/serialization/_fs.py,sha256=dd58907a832626a2bc747b09764b66048eb7eb610a4c7320c87c16647d7d4d3f,5991
spectrax/serialization/async_manager.py,sha256=8cb35179fc75bfeae01393b2f36812127bf2e4ec837f294435a618d88328ccba,20041
spectrax/serialization/checkpointer.py,sha256=3aa84d7e408a4ef933d1024188c9dd1e3ddc8c58aec93a2a1239872d54bcfda4,18679
spectrax/serialization/fsspec_utils.py,sha256=a3cbc54a28392d5df91b6bd410bb2504bba2a19b0f13bdd3cf1e3f318dba1950,655
spectrax/serialization/serialization.py,sha256=f309252b5b411946898386c74d07a825772480db88706c0dc9c242305b9ec354,18404
spectrax/sharding/__init__.py,sha256=6bb2799086c78df84722aa359ee988b566ac841e57a27e1b039c96608b112c61,2556
spectrax/sharding/logical.py,sha256=49e9e68938191c14f2bf56da33f9c3867170b9e8812a64dcc48d1744dc89e736,3198
spectrax/sharding/manager.py,sha256=2651f4c4628a7b5754a594fdac9d089bebbef1bb05459f47ae0a698414f2c4f3,40570
spectrax/sharding/mesh.py,sha256=f7c8cb1a0ff46c4ccaf9a9e357f6c0ce6c43ccf546dcd2e1c77acfb875b02681,34298
spectrax/sharding/partition.py,sha256=0653d4d8e698a99ae40bf3c843a1920e898508851a85c8b75e16ab71b276f699,52248
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=04ed31446f25660f4cab5910eaa50aaafa867d6f800336d5d649ed86bab7edd8,22964
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=fae102fab7a643f0b75d94e6ca54526a8d085255936da7fa49d55bdd0795d4b4,37549
spectrax/transforms/vmap.py,sha256=597d09d29e9d1757424cebae952259452787be6c490ca47b13f1945138bf47eb,8746
spectrax/typing.py,sha256=e5a4e56cbd780e4c7069beb4ca60585e34f5ce10976f1edbad56b74e11f679bc,924
spectrax_lib-0.0.4.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
spectrax_lib-0.0.4.dist-info/METADATA,sha256=d54b0f279f7bdba506215a561f29add64d0a5b417323c10277fe5acdf1249a18,28389
spectrax_lib-0.0.4.dist-info/RECORD,,
