spectrax/__init__.py,sha256=1e7db8fe81db8e2391abb1ce7fb5655f2d43f184901eb297a5950703f11a1839,8401
spectrax/_internal/__init__.py,sha256=9d232efd43cc5ef5b0e521f6ee4dbb270799dd83ca68d4da2ba267a070a561aa,493
spectrax/_internal/dtype.py,sha256=b1752c73adf65879b89360d2860af581a5ff14816553d1aa56e5c8313153ecc7,2452
spectrax/_internal/logging.py,sha256=521e39d9662b2cdfa47dd6b5359b46f03437bbf5bb1b580d5918f4f7cefab81c,15034
spectrax/_version.py,sha256=f01737228bd2eb2e0ad6878b25c1196300de6e1497533b54cdf1d78c4c7453cf,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=c7d3eb22d99a9b1c857fc1e0e7f605bcb6f62dc735e6aa8eba83ae2117a21dc0,1738
spectrax/core/containers.py,sha256=dc18ebee9d90fd67bd8bd7e23358fd66b9f7e8405ddb328090a01ccdeb4b0881,57646
spectrax/core/context.py,sha256=1419d4062a44f73a36649a1a35a0c31fea1146a1302cc6a228c0b28530b7179c,5858
spectrax/core/errors.py,sha256=bee9795c3534fc2641519ff9c0f41229fd47a27e126267abe66fd39d4dd21827,3379
spectrax/core/graph.py,sha256=3f3e0473dc9fb48356f1ad819e6b2fb3a94caa7e82852492995baa616f0de94c,36626
spectrax/core/lazy_init.py,sha256=80808ff92a8e8f53fd7cf81ee841257e0338f2fbe0ee219ca5851734e3062308,1569
spectrax/core/module.py,sha256=cf69c4bb2f94723c5c5b28a768051fd4fff709f0c035e8d15185b04b7c657cf1,54911
spectrax/core/paths.py,sha256=17e88be5419c67dc18b655dd83a2cffba4c20e27faac8f5678c9430cbfcf5b59,3963
spectrax/core/policy.py,sha256=1c8cbe17121a526f4dbe799a35762b3477028e8803ad6684fb6d4abd260a6992,4239
spectrax/core/registry.py,sha256=3e4f8ee9ff9adba7814e76b9581988ac3ed8d0e8df1fb0e7269c5a754ac0620a,3628
spectrax/core/selector.py,sha256=29cb53da43ff483c97b94986238e5e50c07363688a49b18798d630ca20111df1,26743
spectrax/core/sharding.py,sha256=a1797b8f5f5aa9294f4622eaca1df0434eae922e6933cd3d420762a3447fd108,7098
spectrax/core/stage_assignment.py,sha256=71996d9cb2546ab9734cd0c30572f2d694c29b150e9c9f0a94ee052b42679867,8098
spectrax/core/state.py,sha256=174fe4412824665aa2d1eab800431466fa13774b82dda0c93c1417dad203a611,30220
spectrax/core/static.py,sha256=ffb93ff3ddc58f849d5c5d340c379b4a9b54d9b29e8bfff12e89fb2e128b6b99,2810
spectrax/core/variable.py,sha256=b637ca6cc9b207142e83d6e984e8e59dc38795bd241e5a3f7e66f108cbe28b4f,45371
spectrax/functional/__init__.py,sha256=9dd61b9f0110ff526152cc717d0b5b99440881e58f23d040e99dcd37a6a81c80,2297
spectrax/functional/activation.py,sha256=1d86e3fabaec3c0f24ed9ae115a1bc7ac95ec67d81964ad9f3c52bfa3372fedd,8656
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=f967b0c7de046b0946d1686bd5b5b4fab6ecf442eb4a132df920c158bc518bf3,1278
spectrax/loggers/base.py,sha256=ac7e5e995992a337173dc4cc2feeb883da21a111c80edf3960a7ec62cfaf9f34,11943
spectrax/loggers/console.py,sha256=7cd7a5455d89a5d93c8bfa4f6358d432e3f751266c910f4cae1b453c3428719e,4733
spectrax/loggers/summary_writer.py,sha256=8112e851a5bea36d5768fe2992185c475a71687742be8e0af691879241b4005f,1100
spectrax/loggers/tensorboard.py,sha256=854e77aee76ad11a2e6ecc50fe5779cdcf08853589c6bf6bbb595ca0f591d33e,18722
spectrax/loggers/wandb.py,sha256=5b34ca0b197dc50fba641fa67c8d5fbe75e03080bebe82c1f1c53458358b41c6,4220
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=52cd3e540c62471b4900672865d78dba4766c9031d7d804b4a73d82a5749af80,2762
spectrax/runtime/mpmd/__init__.py,sha256=a01940affe0418b45d81571ce621c9ba86a173a785dffe0fcc35fecb3910ed81,1653
spectrax/runtime/mpmd/compiler.py,sha256=999450a8c9ecaecfabe63f2248688a23ed9292a8c5cfa4fe10886d964b0de6ce,15884
spectrax/runtime/mpmd/markers.py,sha256=171864819175038685acd06004ff7f733ff32375f3e01ee7caa9b16c281c8047,36907
spectrax/runtime/mpmd/per_rank.py,sha256=d8685a7120861ae9a26878ce2fb7e1f66e55f004c6728cc66bfc75e383f6366f,13991
spectrax/runtime/mpmd/pscan_compiler.py,sha256=b7fb8bc4c6eb1cdb339177459a68ba1d9b896d4baab4820d1497e6fc30ba2e32,52315
spectrax/runtime/mpmd/runtime.py,sha256=303d3d2ef291a4709dc0d2faf18e577362dab150d0fdc4bb5c960b1898088660,278902
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=75b68110e735e1b8f9782c15816a3678c2c425d969599b6664d2e870c647973b,2619
spectrax/serialization/_fs.py,sha256=dd58907a832626a2bc747b09764b66048eb7eb610a4c7320c87c16647d7d4d3f,5991
spectrax/serialization/async_manager.py,sha256=88417ed2a568210e0301f63e423d3c0cbdece3c7995c3c65c185b65a0b669e28,21801
spectrax/serialization/checkpointer.py,sha256=b6ea79874e4587a1656a4c05aa17ead68e6d1b048eb7908358872a98759bfbd6,20663
spectrax/serialization/fsspec_utils.py,sha256=37d151fef62e4289d02965aff6dbc72a74ce4bada098c95d6accf507bf6abe91,1049
spectrax/serialization/serialization.py,sha256=9f58021a794181ece619df3ec84f112452ae4d5a693bf6e251c486145ade279f,19243
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=e14130cf24fe326f58c3a81e25ffed048616c43f04078cbbd6980248c3929b10,27948
spectrax/transforms/eval_shape.py,sha256=15c1fcdb018511c67580b80f258af9ac357ceff14112a633b37126898acd0d95,2716
spectrax/transforms/grad.py,sha256=2a7521c790e2a6d9ee470f9960c44d22aa52126a5a3ae61e051bca36e62b84a2,44931
spectrax/transforms/jit.py,sha256=53eb7ae5d04bc2aece044a3fa6fa57163b20128e725ce4b83ddcf987a53a5c94,26370
spectrax/transforms/remat.py,sha256=244047a2892b17f391f053f4fc2fc38c4830a6e631d5489dccb6d4f0e8391975,13808
spectrax/transforms/rng_axes.py,sha256=8d2988cadd3b7066718c1593cc1a1303ec17c6a06a271f89e413ab9b4c70a19f,8309
spectrax/transforms/scan.py,sha256=df7b4ff8ea79e369fb98d9a745f480af85788ebc6d91294170eba4400fa7f330,10055
spectrax/transforms/split_merge.py,sha256=803dd0559ea2a5b16e0c8f2e452bfda1ff2fbf120fe0b154bdcf2d04e848d165,38678
spectrax/transforms/vmap.py,sha256=597d09d29e9d1757424cebae952259452787be6c490ca47b13f1945138bf47eb,8746
spectrax/typing.py,sha256=e5a4e56cbd780e4c7069beb4ca60585e34f5ce10976f1edbad56b74e11f679bc,924
spectrax_lib-0.0.5.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
spectrax_lib-0.0.5.dist-info/METADATA,sha256=529227e0ff5b458996e2a2c3e1c5c6986ff7c577f0d28e7591f8166a4370093d,28392
spectrax_lib-0.0.5.dist-info/RECORD,,
