spectrax/__init__.py,sha256=180f8b822a6da310308410b2f328b4b87e9bfd8426a2e01f7ab86115de4870a0,8599
spectrax/_internal/__init__.py,sha256=9d232efd43cc5ef5b0e521f6ee4dbb270799dd83ca68d4da2ba267a070a561aa,493
spectrax/_internal/dtype.py,sha256=b1752c73adf65879b89360d2860af581a5ff14816553d1aa56e5c8313153ecc7,2452
spectrax/_internal/logging.py,sha256=7cb264d307088ad1ef3f25e3faa083b85e1c839f1404bf1cae6d5ad60a85acb8,15064
spectrax/_version.py,sha256=865c5fbda46cd98cbbeaf67be1e0113a62feb96d0cfcf9b275e4276b5db9155b,298
spectrax/api.py,sha256=1958effce4c5b012154720efc8d6683d550bdcb5f76f44f9620b376e432c2f87,19510
spectrax/common_types.py,sha256=5ba851279be4f179374a7f7d99dbac32c42e15035b4a1828ce64b297487a45e1,13415
spectrax/contrib/__init__.py,sha256=ea0065ed5f29814ee7061c102ca812103083c194c9a39e1b77944baf03ec0ac4,757
spectrax/contrib/optimizer.py,sha256=857e34d692db6d82ef82c2520cf53fda39bf896218dfe3d5c240017212ad0a55,17832
spectrax/core/__init__.py,sha256=443f2374f444df7d61fac8385df74ab15ecb732a38827cf83f4f6d6f5b1891ad,1352
spectrax/core/_typing.py,sha256=27aa23755898f65acdd0f31bc52cdd00165954f922f243a0b1240ce97d3f1d6b,4868
spectrax/core/_weakcache.py,sha256=128b5cccec04d0864ce1ecc60bf9c6b5b45182a8d89caab16f629b85af8b1c31,1727
spectrax/core/containers.py,sha256=8046acf5df8ebadf0b7d1aca2fed39928079777efd4d8bfe390f10df284cfeaf,59654
spectrax/core/context.py,sha256=9ae1c8f604bc49f4b64aa1c4288ab7a5de1ab9d3e340c5568f5d8e05a33a4b62,5870
spectrax/core/errors.py,sha256=bee9795c3534fc2641519ff9c0f41229fd47a27e126267abe66fd39d4dd21827,3379
spectrax/core/graph.py,sha256=552e8b394cf75c740562605c4c5f476fe2ac57ad7f0b3ad3dea18f09ce941fe8,36645
spectrax/core/lazy_init.py,sha256=80808ff92a8e8f53fd7cf81ee841257e0338f2fbe0ee219ca5851734e3062308,1569
spectrax/core/module.py,sha256=dafd8ec60bdeb00d65307ae2adc9c4110943b721b08ef8961391c0caf3e110e2,55054
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=123b359e5dfba423a8cbd94d61ffb466995f1c6aa663a484b32b7f2cbe13da5c,26751
spectrax/core/sharding.py,sha256=a1797b8f5f5aa9294f4622eaca1df0434eae922e6933cd3d420762a3447fd108,7098
spectrax/core/stage_assignment.py,sha256=6db6bfca150130714f8017210f7c01b6f7035f86fb5455f784486ded9ab83d5d,8087
spectrax/core/state.py,sha256=a61296c8adb6dbc674f99cacab8015aae9b6f7c751e584cef5711f8d9066bb37,30516
spectrax/core/static.py,sha256=ffb93ff3ddc58f849d5c5d340c379b4a9b54d9b29e8bfff12e89fb2e128b6b99,2810
spectrax/core/variable.py,sha256=cabfa78bda3aa9b91a367ecd2096ff1a914ed8d84f6af604b48fa9530ae24d69,45551
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=33c6233ccdc2ca3b9e89d3a36e81e0ccc73a7c8137243584b3560559a7501f28,2634
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=c8466cfdc9c533f2361ed104bf7d7088291c32987b7be2b683cd7de8233fbfb5,5246
spectrax/inspect/summary.py,sha256=a34776eb980e0317aec82551e994ae09a64b870b2a734ca8144373e8792da1fe,3771
spectrax/inspect/tabulate.py,sha256=490f7fb5fd2c3476ef1ccbf9539602478a8786aeef7bf001bc91a0e820564120,8435
spectrax/inspect/tree.py,sha256=ab5d0e832aac528d837281dd3a178b30663068537f858691bbc7bb1e8d5ad7a5,1982
spectrax/lint.py,sha256=11023ff9768dd092fd2be42135d6d5200905b43575144fb7714e92bf6c18b6df,3133
spectrax/loggers/__init__.py,sha256=f967b0c7de046b0946d1686bd5b5b4fab6ecf442eb4a132df920c158bc518bf3,1278
spectrax/loggers/base.py,sha256=7ffd47d4f79b1a960075885d8162ad4405d1d47068e7bcbcca5227a17b60c148,12050
spectrax/loggers/console.py,sha256=b16fed2414afcdca7cc43659e114545265dd3f3be6e576f13e882ea7d11fb1ca,4727
spectrax/loggers/summary_writer.py,sha256=b808c1eb4d10cf6aeb6abbbce27cc908c61e29e01658523a375e0506d7929c36,1080
spectrax/loggers/tensorboard.py,sha256=0fd243b8b5c7b9a8ca8ad712808f02e1e8f448ac56edf5f3a4c143cebf6f1615,18831
spectrax/loggers/wandb.py,sha256=36bc31bab1574ccc97756e2838ceb6301977b16b0910a5fb9b7780432906fea3,4215
spectrax/nn/__init__.py,sha256=9329c3964c7a1875efddcaf9d91278c35c5ad86fa8fe63f7a459e4e683792e22,4016
spectrax/nn/activation.py,sha256=eac8aa25b45c9594f940f933ecfbcccb873fcc77437e13b238da8e4996e133aa,5757
spectrax/nn/attention.py,sha256=d047ca70c714bca5f92f387eee9c61ef87030a81e4e6235b74d32656ab748afa,15786
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=8d7e9525b810157c72f570053e4c17396ec45e0d1de00c247d8cdba6650d5033,34272
spectrax/nn/identity.py,sha256=0194b73bc7f1a149aff6083500c77059d27b0ef68edbf5349be6219e29fabf89,1719
spectrax/nn/linear.py,sha256=41de95d45e54bc73b50baf6b995053920093541261a89524a1b251bb4ececde3,11040
spectrax/nn/lora.py,sha256=e2eae95e5dfa0ee411f3c923b4bcae9c7d7a616723098597cfb0aa94c6958b88,15037
spectrax/nn/mlp.py,sha256=f445fca57565114d151e59b5ebb8f65e9679ec87b959a2d01b55db76bdf4427d,5291
spectrax/nn/norm.py,sha256=1bb22d02ce009ab0b0b50bc8fe1859b6f92f5fdd33501ce0d4a5c6b63a5fbc03,18933
spectrax/nn/pipeline_sequential.py,sha256=dd5fb76ac04ae66d6e893baf6b579225bfd53963995adb56232b5f0fb11e9036,5877
spectrax/nn/pool.py,sha256=1cab8e28fa11cb71b013eaac692fadd7a4d62795c3c345b122b7187a2124a6a0,9458
spectrax/nn/recurrent.py,sha256=14d100b21bf41ce62a70206379307ea89141bcace7a12a947ab3c0b37e47a3a4,33778
spectrax/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
spectrax/rng/__init__.py,sha256=5c721a22a1b3dafa126991902230312321d51a536e16891c7b7930eb4b29e627,1422
spectrax/rng/rngs.py,sha256=4e879cdd03e91a920b8b7989d064fb88c5fce0cf29d8cc276483addfa009debc,20021
spectrax/rng/seed.py,sha256=a82a825acadba997a9f29e152f1a313f04ce9581ade5f5a1ad40894d978335c3,3080
spectrax/runtime/__init__.py,sha256=ca987f0a7573e664f3e20d2317ea104ff9d6447e89d65d809979754fe3188b6f,2991
spectrax/runtime/mpmd/__init__.py,sha256=b7c4f541e15cc8fdbb87d6262ab37d014d0648a6a1f4396e83f3634879b7cfc1,2190
spectrax/runtime/mpmd/compiler.py,sha256=38f7d5b3943994f80912bc58ff690e9ae6ea600b9d0febbc6924d5670f994b04,16110
spectrax/runtime/mpmd/markers.py,sha256=174652d92c187ce2ff715ce5ab8fc62a53ee85830003797b2418f0c27e7f1aa1,43234
spectrax/runtime/mpmd/per_rank.py,sha256=774a4a95468e6c09ef3a5f7ecf60de987994e9e38399b327c79b049fa3462789,14497
spectrax/runtime/mpmd/pipeline_executor.py,sha256=7a747393d953ee2d4054e920331d140dcb8d79e0dda0104361513d93961120b4,49012
spectrax/runtime/mpmd/pscan_compiler.py,sha256=85861a4100b3d3562a8062b80527b5548bc5b5860306d4ea382461fe6ae669f0,68924
spectrax/runtime/mpmd/runtime.py,sha256=0f6cf528ca55ecda88c7df16e7f0b034c16b1619de41200704c19e8cc6681909,331324
spectrax/runtime/mpmd/treduce.py,sha256=07cde54d15d6f252d6aa44c258ffa3bf9f6236539b17590559e7d6bd597c4326,20586
spectrax/runtime/primitives/__init__.py,sha256=0978c4c75a1f30dee61116a8dc6a8e903f0c9b1be88c4d4a177605f2a30dbd7a,1059
spectrax/runtime/primitives/boundary.py,sha256=b1667734488e662899dd1e0161de15553ad09b86c23c250d96e79b2f96760d58,3301
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=2beb41b951195f99bb27c698e9d2725b98a263168782b44d4fe5fd8e614b94da,1094
spectrax/runtime/spmd/api.py,sha256=5660a520ead0be42f275f68f1c949dede318d7ebf6ab818ae521f597c8050cd7,4630
spectrax/runtime/spmd/hybrid.py,sha256=6b6f6c90d6b9192660ffac7e1e12e95e149763c3ee046eb2d58929bff43b758f,1013
spectrax/runtime/spmd/runtime.py,sha256=18ee5e3b1b3a766329afb1f4219754976b0e9e497a06efcd1745bbb9c8da8782,19445
spectrax/runtime/spmd/shard_map.py,sha256=0aed1404e34753fadd121925d1eb823464fab5bbb3423ffb32b6e60f30bd64c9,33206
spectrax/runtime/types/__init__.py,sha256=289e22e11c5e6ca492547ec7416d8143f8c6bed5357eae86f58dc44154b9846c,1377
spectrax/runtime/types/array.py,sha256=306f71c96881fe6c465f9b4c93393cb27c5421c15c513375372517fa0004fb40,12945
spectrax/runtime/types/mesh.py,sha256=4a7940cf29bb1e9564aff2c767b507b364c6d236cadee747e94978cc16b7e5a6,8809
spectrax/runtime/types/stage.py,sha256=fb52239ea331de872014c51eba1b1fe3fb037af81da144320d855715178c680b,3494
spectrax/serialization/__init__.py,sha256=39096083a8acf361cf4475497f5fa8cda322a44215a02bb8e2ab6ac78cc394b0,977
spectrax/serialization/_compat.py,sha256=a3050a4bbaa027c8e8ffedbc4cd54d12e7b8eec5614dd4be5e71d0e45e0126c6,2739
spectrax/serialization/_fs.py,sha256=dd58907a832626a2bc747b09764b66048eb7eb610a4c7320c87c16647d7d4d3f,5991
spectrax/serialization/async_manager.py,sha256=bb747a36f4bfdd0e453bc3809ef292b74610d37f2e295ba69fdeaa8961d5c5d0,32134
spectrax/serialization/checkpointer.py,sha256=2384e90af711015368217c475b64377485827499c7d9846a3b4dd794ffa43617,21262
spectrax/serialization/fsspec_utils.py,sha256=37d151fef62e4289d02965aff6dbc72a74ce4bada098c95d6accf507bf6abe91,1049
spectrax/serialization/serialization.py,sha256=86518b67794fda73c222515bc41c5ff850cd6644b340ff06261e18795672f2d2,19259
spectrax/sharding/__init__.py,sha256=6bb2799086c78df84722aa359ee988b566ac841e57a27e1b039c96608b112c61,2556
spectrax/sharding/logical.py,sha256=49e9e68938191c14f2bf56da33f9c3867170b9e8812a64dcc48d1744dc89e736,3198
spectrax/sharding/manager.py,sha256=4db5e60ee86c0fcd472e12c5f9fb3394b3240982067417e7265b7a37544f5255,41254
spectrax/sharding/mesh.py,sha256=dce644aee87310df0aa24b2389550043b2476949ba031ce479d969fb8d660117,34318
spectrax/sharding/partition.py,sha256=998f80149515d36bc1c52919ec4cec0d9282b0bfd0781ab04e7beec997affd23,52803
spectrax/transforms/__init__.py,sha256=ce80363cf9d80f96f22d463c2139090adfd274eb99282731dd3aae79772f01b0,2007
spectrax/transforms/control_flow.py,sha256=883feb4303ed8d6d391cb825c238017bf205eabc2ec53e5f91d9cb8e2eb8c0e7,28051
spectrax/transforms/eval_shape.py,sha256=7207ea4f5d2ba8eafebf40da87aed330710412f21b6edd2ea539dbee73d4bbef,2757
spectrax/transforms/grad.py,sha256=dab33c13c0c08a7b46a39bda15ce9c39b3e67b92ea92471482d3d3469322ef09,45549
spectrax/transforms/jit.py,sha256=a4082ee40fcd3864069a72efdb5cb8c9a271ad03550b068a6104a2929375ad51,26677
spectrax/transforms/remat.py,sha256=18555d180720d8d47a37be3ea1f598e5ec56b7c0ebda6a7af7a858b8124b5e9f,14008
spectrax/transforms/rng_axes.py,sha256=19d6947d884d5dd69c88847e0d2955886f92d90d8f3ac90f8c9bb4875c0154a1,8288
spectrax/transforms/scan.py,sha256=b51de9bdc6fd706c2a5d3306a7250f0e5c913fb7fb09e603864436d0413b1f9d,10072
spectrax/transforms/split_merge.py,sha256=5ba71cf428704686309bf7717acc49c70f1ddb84f233d0517216cec3f637184e,38959
spectrax/transforms/vmap.py,sha256=a15d01d7a88319317090f2a0e03ac1715b03fcea2ae98b06a008c84fe2cba7d1,8866
spectrax/typing.py,sha256=51b43db8af95fa07812725c96152bffea96d4261ecb570eed2ba392ad1f271fe,950
spectrax_lib-0.1.1.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
spectrax_lib-0.1.1.dist-info/METADATA,sha256=f77009251457f92c4dbcfb32389b03c5d7638b6ce9ff9090207801a092a5bae0,31716
spectrax_lib-0.1.1.dist-info/RECORD,,
