spectrax/__init__.py,sha256=17aa926a83a49a6da9bddc96951f35f9555905ad601c27bde72db9259b485e0c,8729
spectrax/_internal/__init__.py,sha256=9d232efd43cc5ef5b0e521f6ee4dbb270799dd83ca68d4da2ba267a070a561aa,493
spectrax/_internal/dtype.py,sha256=b1752c73adf65879b89360d2860af581a5ff14816553d1aa56e5c8313153ecc7,2452
spectrax/_internal/logging.py,sha256=aa9666a060b071be442872eedacd498ebea3e37d75a0866b896bd62b655d73ed,17693
spectrax/_version.py,sha256=077a20a23db945eebeba65acc9d8c163b905ed5469048811a52a32ebd35e1cf8,298
spectrax/api.py,sha256=8bc9f7f5459847d621ce4d094ba6ca74f6e74d2f527863e6a881fc60cb0fa0d2,20602
spectrax/common_types.py,sha256=5ba851279be4f179374a7f7d99dbac32c42e15035b4a1828ce64b297487a45e1,13415
spectrax/contrib/__init__.py,sha256=ea0065ed5f29814ee7061c102ca812103083c194c9a39e1b77944baf03ec0ac4,757
spectrax/contrib/optimizer.py,sha256=d56c55cf0356fc9486f46569f06db4c425cce27c160a73f979ccfcd00e6b19f7,18055
spectrax/core/__init__.py,sha256=443f2374f444df7d61fac8385df74ab15ecb732a38827cf83f4f6d6f5b1891ad,1352
spectrax/core/_typing.py,sha256=30f582b03273d1bfec79ed72d1f8e65f44f98f9fe9baa7719d9ce912a41f6507,6086
spectrax/core/_weakcache.py,sha256=128b5cccec04d0864ce1ecc60bf9c6b5b45182a8d89caab16f629b85af8b1c31,1727
spectrax/core/containers.py,sha256=436153cdab783703cbc4a4b2d411d1c604aefb4b7181720d79c2b29cd5ce48f5,70887
spectrax/core/context.py,sha256=f0288fa92fea61300c84cf59db2a8baceda6066637c8bfd47da6712fce4e48b3,6126
spectrax/core/errors.py,sha256=bee9795c3534fc2641519ff9c0f41229fd47a27e126267abe66fd39d4dd21827,3379
spectrax/core/graph.py,sha256=8c1a6c00a7c5b2bacb7b4d4acb0c6c6155ea85a9a68d5074d08be7a959bceff6,39354
spectrax/core/lazy_init.py,sha256=a9755ef3de13af3f2a5bca2c0eecf1e7274552aac9892ef43ac37cbcad8805f3,1816
spectrax/core/module.py,sha256=25cf6ef890b0463c67991c139c6a3b0d37f19808b5794cad43ce2a297706f63b,58019
spectrax/core/paths.py,sha256=63c6a0ecd74a1b2d1d91436238474bf9d0475b3b240f5fe3778d02f41d4d7858,4076
spectrax/core/policy.py,sha256=233b077cea6af3c7adc24fa94357e581a5d15e83653d8732e29152a439d7b39f,4316
spectrax/core/registry.py,sha256=3e4f8ee9ff9adba7814e76b9581988ac3ed8d0e8df1fb0e7269c5a754ac0620a,3628
spectrax/core/selector.py,sha256=4dbdd28ee985b20ad6008995890543e3015e7595d79ebe408ebd51c4bd7a58ec,29197
spectrax/core/sharding.py,sha256=a1797b8f5f5aa9294f4622eaca1df0434eae922e6933cd3d420762a3447fd108,7098
spectrax/core/stage_assignment.py,sha256=6db6bfca150130714f8017210f7c01b6f7035f86fb5455f784486ded9ab83d5d,8087
spectrax/core/state.py,sha256=24b6e432bbdec9fd784de5f2c953f3bef8efa0c84465be15cb854ae0d12af8c5,36680
spectrax/core/static.py,sha256=d0211bf79d143dc809159fb645bf4b8023ff752a4a82eb8c7911566b406a4aa8,3176
spectrax/core/variable.py,sha256=eaf623e7c310fb884e35fa15f92188b789841c7834609b557c4d50d8cb23c7f4,52059
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=c7dac8f59c3b14afaf5b77fd6ffe5cba2e996c4b3df15cdf992781a24ee6a332,3661
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=cdc078ac6def5c2d9054c75ddea0d7193049ba72851a00ec3e9e6e7eaf20a3b4,2574
spectrax/init/kaiming.py,sha256=8605342b21475fd1e2e0258f4095a2275e370e5a3ed1e8b8fdd0fdc9d1ee544f,4625
spectrax/init/normal.py,sha256=8b4937a3b970fdd0501249db44d582317bf3a05050cb000087bcb1c5ccd2f020,3131
spectrax/init/orthogonal.py,sha256=985d8154ab82cb5282fa74fc01acb1f6e23ae28767de8865101aabdfdeddceb3,2936
spectrax/init/uniform.py,sha256=88e50e4f798dbdcad611f0f0043a621eebec297cdc921ba6c576bddfc00b1bb0,1476
spectrax/init/xavier.py,sha256=f9e866ff24182ed814b657310ec3c8fab21f5869a041e2b8e159203fd165b0c8,4236
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=9bcd1afa9c03443d96187de1b24f2603a535aa9b6852d1ae410d9f6c0008c574,8727
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=ddf82c78ca5e4cd7e30a66d577ee00e36102dcee032e4dfbd818903b2815c67b,15134
spectrax/loggers/console.py,sha256=46b53c88e15c2a0d1775113b5fa5afc1c8b189f8c4ae9558d69cf4f232c3eaf1,5170
spectrax/loggers/summary_writer.py,sha256=b808c1eb4d10cf6aeb6abbbce27cc908c61e29e01658523a375e0506d7929c36,1080
spectrax/loggers/tensorboard.py,sha256=5b1a0bbde0e1b4d5ac5c263e35216f4f1fe213f4af83c2c344de547908afb6e9,18955
spectrax/loggers/wandb.py,sha256=36bc31bab1574ccc97756e2838ceb6301977b16b0910a5fb9b7780432906fea3,4215
spectrax/nn/__init__.py,sha256=e72b21af3749558204a8632df13d2512485cc1b0d395c499b0fe976337473eb9,3944
spectrax/nn/activation.py,sha256=eac8aa25b45c9594f940f933ecfbcccb873fcc77437e13b238da8e4996e133aa,5757
spectrax/nn/attention.py,sha256=cbb0a973dd1dd6fb4f437fa457e315c83b787e91baa5bc1d441b553f245383c6,16679
spectrax/nn/conv.py,sha256=f4b467d1e4402cd999b5ee7ee2ba75fe2f1a3dd574afaeec4960a852616afb24,17858
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=273ba6c21019cfe696ccafaba789c763e6c2985a2ae1b00241e273b1b1b19bbe,34507
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=42fc7b7ae5c1b140a4df2bec822b12c4f1f890be9c42e146b8672819f68afcf9,6223
spectrax/nn/pool.py,sha256=1cab8e28fa11cb71b013eaac692fadd7a4d62795c3c345b122b7187a2124a6a0,9458
spectrax/nn/recurrent.py,sha256=038c0b0f998d78e481227a2e3ca1733e131637f8f5c55a1b43910483fac6adde,33754
spectrax/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
spectrax/rng/__init__.py,sha256=5c721a22a1b3dafa126991902230312321d51a536e16891c7b7930eb4b29e627,1422
spectrax/rng/rngs.py,sha256=6ddc80abe77f4437e932ba837da73ea9829727bd29f12f4a07b9e560b80532e4,21313
spectrax/rng/seed.py,sha256=af223b403cced1f3d9507c0b48b5ed6a836a6248e1b4a1d75005d4c5cd9fd65c,3272
spectrax/runtime/__init__.py,sha256=ca987f0a7573e664f3e20d2317ea104ff9d6447e89d65d809979754fe3188b6f,2991
spectrax/runtime/mpmd/__init__.py,sha256=b7c4f541e15cc8fdbb87d6262ab37d014d0648a6a1f4396e83f3634879b7cfc1,2190
spectrax/runtime/mpmd/compiler.py,sha256=0271e2130a4d229b412a043eb6489982c9e7e93b1b17297bbd02c7590ec906b6,17485
spectrax/runtime/mpmd/markers.py,sha256=74675fe4c1702f0d1d6fbcdb8e6edb0c7963ae576de8fb3770beed23b2e0bf44,53295
spectrax/runtime/mpmd/per_rank.py,sha256=ecd096272cff2a31baddecf9c8558fc91193fbb96a378afbd5a96d197c0fb06a,14995
spectrax/runtime/mpmd/pipeline_executor.py,sha256=f6bccdb01313c1d13abbff7e19d710c1be2d8d3b568c9d10a37ac228c42ceb5b,52864
spectrax/runtime/mpmd/pscan_compiler.py,sha256=bab2fb1ee095e8f7ad19e3afdf7b5e0aefc070c8f7313a161d26c2d17525d239,80688
spectrax/runtime/mpmd/runtime.py,sha256=ff66a23f684c9ded5909129c4310defb6afcdd2f6aeb06d39e1ad233c1272fc5,365465
spectrax/runtime/mpmd/treduce.py,sha256=e4b128f6d0eef41c0e3715202676f6c7ac28e03bce558b7f30868e19e19e0c26,24901
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=1bfe31770793a67a19e4993760d6280175a52f5b75677e89d53258685d078802,11299
spectrax/runtime/schedules/dualpipe.py,sha256=93a7426d29c773d4acdd538c04526836b73475ebdef3700dc0c3f6fe85f34a3b,14957
spectrax/runtime/schedules/fusion.py,sha256=76abe8ac813667f2524ea3afd033bd405bc65d21a01483c1ca806e3fab8259f1,4584
spectrax/runtime/schedules/gpipe.py,sha256=345c5a83fbb04b2f27a519d4d6b5616352887b5aed85f5bd2805da84b7954bc6,3226
spectrax/runtime/schedules/interleaved.py,sha256=80c28308c634fd0c6a4cfbd1e5193e1b601a5697e34d0449e034c04062d347a4,28810
spectrax/runtime/schedules/one_f_one_b.py,sha256=38660241e12800d0b2ea4c3bb7f573884687c7a31b8fe7025dc8ee2443b25971,9591
spectrax/runtime/schedules/zero_bubble.py,sha256=2336bb870f3e02a60dbce3c5ea2f5665e72455a2ddf7281f94cb5e45d90f6b18,4995
spectrax/runtime/spmd/__init__.py,sha256=2beb41b951195f99bb27c698e9d2725b98a263168782b44d4fe5fd8e614b94da,1094
spectrax/runtime/spmd/api.py,sha256=5660a520ead0be42f275f68f1c949dede318d7ebf6ab818ae521f597c8050cd7,4630
spectrax/runtime/spmd/hybrid.py,sha256=5a19cae25c0719e3bae619c592a9a1663fdb88bdb7c3299b344d9ab206d76cd3,1277
spectrax/runtime/spmd/runtime.py,sha256=ae8dd8772e673ee923e116a120f371ba18056a9e22c80173fd1007173016f37f,20599
spectrax/runtime/spmd/shard_map.py,sha256=b09578cb0d2fbc56954809e1e393ca149f7e413bf1cc4a83a8916398ffd63012,35406
spectrax/runtime/types/__init__.py,sha256=289e22e11c5e6ca492547ec7416d8143f8c6bed5357eae86f58dc44154b9846c,1377
spectrax/runtime/types/array.py,sha256=21dcc3235cf4014d4d57cef3faa83f251e29e441c5bb5c2de67a9e315088dc93,14678
spectrax/runtime/types/mesh.py,sha256=e0a7baccc91ea09c527908cd36687a95014695a5c5cfe0683de12f91e00dafeb,9456
spectrax/runtime/types/stage.py,sha256=fb52239ea331de872014c51eba1b1fe3fb037af81da144320d855715178c680b,3494
spectrax/serialization/__init__.py,sha256=39096083a8acf361cf4475497f5fa8cda322a44215a02bb8e2ab6ac78cc394b0,977
spectrax/serialization/_compat.py,sha256=e418f990cba4fe0d9468269b3b3906cb2c86359432162bb4c7fa5d96fc687b0a,3251
spectrax/serialization/_fs.py,sha256=dd58907a832626a2bc747b09764b66048eb7eb610a4c7320c87c16647d7d4d3f,5991
spectrax/serialization/async_manager.py,sha256=3f6d4a7e0a09fabde56c6ed6d8850d8c346cba532baa3da6e024e13191d592a5,62945
spectrax/serialization/checkpointer.py,sha256=cd3fa442176d370f7a8daa11f496cb170f21f3abd8f2751122e4fa969d41ef45,23151
spectrax/serialization/fsspec_utils.py,sha256=37d151fef62e4289d02965aff6dbc72a74ce4bada098c95d6accf507bf6abe91,1049
spectrax/serialization/serialization.py,sha256=909b765b54076f13bd951db595f38fad52c34aeb4a4161a265f6a6a52d612d56,19384
spectrax/sharding/__init__.py,sha256=6bb2799086c78df84722aa359ee988b566ac841e57a27e1b039c96608b112c61,2556
spectrax/sharding/logical.py,sha256=a8b44a8228dda0f28095f0ba2344751c94b7b67f5c8cb78ec6bb366c86681818,3273
spectrax/sharding/manager.py,sha256=78764f92ebcc4707e41935b57bb720917ad4c096b6e6eaff86ae9c724ffb1517,42718
spectrax/sharding/mesh.py,sha256=e18766efb6b947e7a179c4396957470f993dee22056fb4bf4c95a425794faddd,37649
spectrax/sharding/partition.py,sha256=f74695cb0d7ca44b38f44cdd1925c43d2f2a95c5cfa57a7aed5e2236712744c6,60926
spectrax/transforms/__init__.py,sha256=ce80363cf9d80f96f22d463c2139090adfd274eb99282731dd3aae79772f01b0,2007
spectrax/transforms/control_flow.py,sha256=507d9b154e1fc8c45f4c2e754f713afd50e1632d14e251a08437f03f13d8b616,29630
spectrax/transforms/eval_shape.py,sha256=7207ea4f5d2ba8eafebf40da87aed330710412f21b6edd2ea539dbee73d4bbef,2757
spectrax/transforms/grad.py,sha256=efc8848fbf3b2777871beb038d48c24f6f5faf3512ce429ea15f88dced9da30f,52659
spectrax/transforms/jit.py,sha256=c4297942ef8fd6e14a756f2750d84206fad5d58178e9263171d8490174c06186,28404
spectrax/transforms/remat.py,sha256=305ec4288b2bdfd0cc65b9f5aee1701f5bbfb40774e39a65e854c235771aa42e,14917
spectrax/transforms/rng_axes.py,sha256=b207e781a1d3847e7b2c1cb2878ba312df91f8551e3b7de440047e1cdbc2eea6,8363
spectrax/transforms/scan.py,sha256=a4d5342ebe89f5f53eb9121f34a388cc2898dfce673a9c7c571476262b8b8910,10449
spectrax/transforms/split_merge.py,sha256=5eaec2f9816e4ae6f34604c9ea2f9ec8cc72b39d55f32efe8f852a18509ccd23,42289
spectrax/transforms/vmap.py,sha256=7fda881bdfd74b2b5302a6c5213d1a273d0b6f09f900bcceb34a203aa62e9bab,9138
spectrax/typing.py,sha256=51b43db8af95fa07812725c96152bffea96d4261ecb570eed2ba392ad1f271fe,950
spectrax_lib-0.1.2.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
spectrax_lib-0.1.2.dist-info/METADATA,sha256=11c8e0805d4338b8da3abe829e70ac65b7a5bad2d234f9f8e23ba92cf91e4426,31549
spectrax_lib-0.1.2.dist-info/RECORD,,
