specflux/__init__.py,sha256=745552e9b204d21a62edb720db64935b1435d0ed769d611cebc9a65d4e8effc5,4450
specflux/_version.py,sha256=e7ca9342b6d94fbfc38240cd6211760139d15a5221bdbed1ba7da47eed5f0939,761
specflux/api.py,sha256=dbc9d5f27d7ab2e2514ee420e4e76edf1a9f211cf314135522d5f2601c5fa899,12481
specflux/contrib/__init__.py,sha256=6d97717bf9d780523065438b4bf9fb152c3c68389551aa45881ce90e40973576,884
specflux/contrib/optimizer.py,sha256=9d7f7cef34afa53113a50d4339b35a81801143c829c909fd3fa5491faf6940ee,14978
specflux/core/__init__.py,sha256=37119742863f72a6fd88b8dd6351953b395deb37901d0a54ff83ff0884f0cbf6,1715
specflux/core/_typing.py,sha256=7aac27b85f452438921aa1264209a95027288c5771e1661c60c55773b60ef997,4679
specflux/core/_weakcache.py,sha256=36a0124f7381bda1138e093cdbfabcbc9d08e3ebc7349ad7683c48cb01f4c538,2169
specflux/core/containers.py,sha256=203bebe5862431aefd0974337f57f55ae17aa8e345b3a8ea39e5fb33b31033b8,12080
specflux/core/context.py,sha256=01c06d43ccfab1904585b178ac3f5e3208076cc6f1c4133d042733b616b1c45b,6214
specflux/core/errors.py,sha256=b1aa818b754b0f6a1cf7127f287217d63671d85ad355a36bacc75e269b87cd2c,3535
specflux/core/graph.py,sha256=c3bcef429ddbed0c50a72e8a849f26b226d6df40f50cd026b89deb102f409e0f,32505
specflux/core/lazy_init.py,sha256=3bfe9c5c0c2a76cf01fdf6f254a9f7d0ca4ffdca8151aab6a0ca53496a8fdcdf,1965
specflux/core/module.py,sha256=90aff216ffe2783498a9d83589d50eaf7c0a8bb59989fb95cfa20e7d20678cca,29845
specflux/core/paths.py,sha256=1373081cddf209cc5b140bf22ed6cc944f24bf387000ab96fbe8eb640022b2b9,4024
specflux/core/policy.py,sha256=189e55d07707f22b0fb1fd8c17f1b2cbedc575e819ad1cc7e06d540ca895ab01,3777
specflux/core/registry.py,sha256=fd9ee05868faed38a7739edcc7a3e7367ea4a8f9f4a94affc5b6fa8997d2c439,3948
specflux/core/selector.py,sha256=763a238910b64e4354cad37268e24e3877969ee23d60243d542c0d87e7792564,21897
specflux/core/sharding.py,sha256=91237f9b5c151f2c7daca7fa3171bfaf1582515476d27bb7bd215ea0d0386195,3787
specflux/core/stage_assignment.py,sha256=557aa2ba9ee416332ff0127000f6655ef137b841092e8666ade064fd15db0c7a,4063
specflux/core/state.py,sha256=b8437d2f312c3978f9436da3006f1aedd0208b81b9ce5749243029618acfaa2b,10964
specflux/core/static.py,sha256=23de71c73f5cd5e1682b15e7a3d46c6ffd0446a89143da2f086ddf551238118c,3160
specflux/core/variable.py,sha256=ab2c51f122228667e0b9e5cea7aeecd94bd05c3d4c13d358e75f667000d87c21,29751
specflux/functional/__init__.py,sha256=3b3986a44250f2e9575aff6308f6453bd47d91fdaa6082d2747b5bc7be273f8f,1926
specflux/functional/activation.py,sha256=7413cfce0cb4714dda00e371e3aa70e7d414ee706eef5f6d96db3971426fda49,3732
specflux/functional/attention.py,sha256=4ca962b20a5a528791ba3b8bbce051ee8333287f10f00ddc36fb80d30a49d4fc,2946
specflux/functional/conv.py,sha256=767379245ebe016952dcf770bca3a0910179610f2b60b86c90838586d6cfc3b1,3907
specflux/functional/dropout.py,sha256=399e33d60d8a92d6bc3c7630ea6393a7857755c293a82ade1497dbfd23ae4c5a,1761
specflux/functional/linear.py,sha256=ad7590868d35982bc75646b79bcbbde82fb18b87a1a3065e181455f5b33c4a98,3157
specflux/functional/norm.py,sha256=3d71ac81ac605d35ea956184e15230ac4a90907111fc4932ce1decf2869fbd23,2486
specflux/functional/pool.py,sha256=88dadb74141e9314e7ddc6e43aaca2642ce1c40b9d3ec3acd2f3d76a050fb0ec,3455
specflux/functional/util.py,sha256=19e683956a4a375c1e426728c270da0991d03e3825ffdb7276b17d6b320455e4,1356
specflux/hooks/__init__.py,sha256=baf56f1d20d725fbb217f2f9c6ba988dbb9663d8aafff7121a155e9157b971b9,1146
specflux/hooks/forward.py,sha256=e81b50213e394cbb177b9635bf70d09abe83c9b4a553bfbb447a54377271f563,1789
specflux/hooks/variable.py,sha256=a6a771c51f4ac4c5da38a0640d3cd1cdc9525d430eae30e59c7ddbf59d9fa669,2058
specflux/init/__init__.py,sha256=a003c8da1ff5f0d987b2db222c64b5e8f9a956b1050109c636115fbd5ad25d60,1315
specflux/init/constant.py,sha256=b00059450e7fc2f809a551f3aedf2e8f3f5ff56a51653d1c0c533be204d47665,1562
specflux/init/kaiming.py,sha256=2b55765237bcd2e097d2d22531f7a6e2ca1a9855a09466580b19c24b2883bd9a,2569
specflux/init/normal.py,sha256=6b886284530c40278292f3d3d68dc8a0996b14197103dc96f7fcd873b6dd1189,1680
specflux/init/orthogonal.py,sha256=27297d5a811564707d96e638a1c2a6ee64c0a5cd1104699731501364c8075b21,1819
specflux/init/uniform.py,sha256=435f92237620ab38aeb8abbc29d926a391c2ade1cab7d95eb2fa26b31f5ef079,1149
specflux/init/xavier.py,sha256=00268ff71235982976e6246bda4704737f38267bdb599bcec1bb6ee9dee4a124,2760
specflux/inspect/__init__.py,sha256=3dc6e17d39da007a3181b7f4a999ab76be119bbcf0f17ce99a9db2f346ae1815,1072
specflux/inspect/counting.py,sha256=806dfba1ea3c588c5c23aad9c6c9185ba674f6a1418046c6c41acf60624fb593,1077
specflux/inspect/display.py,sha256=aa842958b4b1fbff89b23d362a0c662c4f91bb3d6b5ef92c65bd430aa249807b,1248
specflux/inspect/repr.py,sha256=0a9270858184cc1eeb7f6143682c2d618a51d489e1b5022081d3e38c43a175f1,3290
specflux/inspect/summary.py,sha256=a642c5331aafaeff77ab887bf8fca4ba82303e18a8b9d9c9f2dd8579ea3cf573,3088
specflux/inspect/tabulate.py,sha256=18c5c619abb5ad2f5d92a37d269ae9bdae0d5b2be470499b157b9e3a1dfffea3,4918
specflux/inspect/tree.py,sha256=ed31e43c0cd0895ed7885907ca17a27913d86e55c50d71921c77b3a362efef88,1645
specflux/lint.py,sha256=49795472ce27027bc1641e2d013346ceb2fceda1fa6cfb76b5ac768fbaa5ac7d,2616
specflux/nn/__init__.py,sha256=7a33dee9746523c35cde40dd4daff41c5aec686952c0c9de96010e5a89b6987a,3036
specflux/nn/activation.py,sha256=95e5ecce35875e034d4cc6cdd4d9c117f1b8caa3fa99601b46a1383de3285b11,2919
specflux/nn/attention.py,sha256=90f083ec6cc589abb12f44f198149fefb3290c0c24b954ce2864d0f2bc33aa12,9667
specflux/nn/conv.py,sha256=c06d0e42fc21985180230273f98a84f2a48d1661593f26cc4d2d3ca20b873d12,8674
specflux/nn/dense.py,sha256=886dd51b0d75cc0e534d8776b5a581da6e03edbaf53f379a1ba78e3542a928f7,7430
specflux/nn/dropout.py,sha256=21bfe4dbc7c1d87e4cce50bccaefc00032069f19f1b39153b5b41b232d17cc2e,2902
specflux/nn/embed.py,sha256=94fd1495dd3cb80caee20ec725a3e159a56090c3fee4ebc5094801fbd9711a6a,3656
specflux/nn/fp8.py,sha256=d3ea2ad2cc918a9d7bb7e043a8452743c16e088867cc7c6ac24ec44d785cd3ac,29091
specflux/nn/identity.py,sha256=d7905c74b8837cb99a71563ba0be8fd26c69834b7383daaa32da7100355ff8b3,1144
specflux/nn/linear.py,sha256=bb3bc83d100ce9bf80d6b7e48f4d146313210aa242f2987533ec2f8a0ea63402,6688
specflux/nn/lora.py,sha256=932c311afaadd519a6f1499afa8a5a54318351bda4fca9a58c46d520fc2c443e,13114
specflux/nn/mlp.py,sha256=240143a1a827c49624184ba5b37ec79d55800be4c6c068bba813d6fb3ac70606,4040
specflux/nn/norm.py,sha256=c379ec5e6190a49029c069205c1e93b3cc5e815eedbfda928a4d6c66b05c153f,12598
specflux/nn/pipeline_sequential.py,sha256=4f43dd3ce187a213000747868e136bae9a6000ce7ce409748a1b45ee5fa80a20,5089
specflux/nn/pool.py,sha256=91c2668daf5cda2693aac80fe9543baf74d34780756c450ce40b112c684bbdab,5026
specflux/nn/recurrent.py,sha256=f0dc5a701f44bbead8750b583cb39a8c48fd94ded593fda244027a1da320c6ba,19306
specflux/py.typed,sha256=e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855,0
specflux/rng/__init__.py,sha256=e675c65f15061f7fe69e0a7ad6edf26eabc07c5fea549783a95168ccd93cac63,1091
specflux/rng/rngs.py,sha256=7909f9017c214e12d4a3408644e8826fa84b4cdf1f9b25d626fea585e76ea5cc,9676
specflux/rng/seed.py,sha256=829c826c21d7f9649f5549b858e4025e1449e79cda8db0f96a48008c81fbde47,2397
specflux/runtime/__init__.py,sha256=70bdb8ef5cb6c6412877c4c348b3b5c52203376649d656f7e0172d01481794cd,2036
specflux/runtime/mpmd/__init__.py,sha256=d6b53cafee9658b51ebe5d4f41b9720dc3ced66a652a204ac1ffa3956a0bffdd,2074
specflux/runtime/mpmd/compiler.py,sha256=2f9d3d5dfe03e18dfff5a07cedca15d0a345e62d7ec9a438bbe3ba1eaec903d2,12855
specflux/runtime/mpmd/markers.py,sha256=9e0f14e39e93b41328069e13f8bdc7339d3a5c9c65b4a162a64fa4e3b298dc3c,15319
specflux/runtime/mpmd/per_rank.py,sha256=7afc15ac68fb3278a34b1fc6307accf8bee6885b380ce10d5074606f51ab4597,13253
specflux/runtime/mpmd/pscan_compiler.py,sha256=5b178c2484cec0f86707af5afb7c4659634d024dab603e9407e380dce0d93344,39031
specflux/runtime/mpmd/runtime.py,sha256=86befb3095b0dbad210ae13ff05ece0616220214773dfb47af4aa95ea2c7e458,125558
specflux/runtime/mpmd/treduce.py,sha256=3d5721e2a52725fb34a965f2b8081a3415d334a57d7233c1c6760db4992b1a5d,16712
specflux/runtime/primitives/__init__.py,sha256=9b0a78f782fa69f1cc2d299dbdbc878d2bd41308cb52171ffde6a891ebd2fef7,808
specflux/runtime/primitives/boundary.py,sha256=00f9b6a5400de5ee7c74f52ee023349b90ec9409cb010b6672494f10d855dcbc,2950
specflux/runtime/primitives/split.py,sha256=b850022d15c74dd0b530e8665dd2b65c27c493b42e4e2c91c7059f52370ff102,10695
specflux/runtime/schedules/__init__.py,sha256=642addec0dec8d12d5b750c5df4bc6b6eb5d32c52abb5cfd76fd9651bb745acd,1371
specflux/runtime/schedules/base.py,sha256=edbafdd4f6b3250fd59c3f9fd54855201b4e74870a8e78510f9c0805c0f16804,7751
specflux/runtime/schedules/dualpipe.py,sha256=e77c7d8271f63df591c6737fe9e98699ff103e1e2ccd53ac2d102a3aa8d2fc36,9999
specflux/runtime/schedules/fusion.py,sha256=0767671317ac7e52f0163caf79582833cc641c50cbda7d62ac4419b41c5c6929,5047
specflux/runtime/schedules/gpipe.py,sha256=96815fdd830fe97176bbf84160c0a17e7578eb56fd1bcb33982fb8beb907fbc8,2813
specflux/runtime/schedules/interleaved.py,sha256=69b127dafdf26d892577790b6663c640923a4b31109d2e12753fce7a746fd981,12199
specflux/runtime/schedules/one_f_one_b.py,sha256=56a9515ed1c77b0502d2a37800a347f005608068a0fdc9e4517cb9b2c9cf8096,7193
specflux/runtime/schedules/zero_bubble.py,sha256=e9d45d273c78864561a48e87cd3ef160c0ee83c795cdffe891431e54111716a9,4580
specflux/runtime/spmd/__init__.py,sha256=4de43f0ef15262f61a641142acec436b8543d7b9fd1922ab37a2f0a4e6b110c8,1033
specflux/runtime/spmd/api.py,sha256=fa52b493f747bacc30ade825e082a7a9094fef371b709ea8faff6dc51813704b,36327
specflux/runtime/spmd/hybrid.py,sha256=6b1440b3292070bea5ffc2f5752bd7b3879d46a2feec5ad413c9851d9b5822e9,12638
specflux/runtime/spmd/runtime.py,sha256=0a0fe47f61647228a7a364facfe6c4d8343eaae420acaae7f66638f95a507d5a,13850
specflux/runtime/spmd/shard_map.py,sha256=3d947d5bdf3b852e756287b6952acdf8a399b8bf1ed54d20e7ad1192b60ff4ec,22030
specflux/runtime/types/__init__.py,sha256=a592013a746d09f76d30f59ee36f2c22a6700aebe83df735c84daca92875a46c,956
specflux/runtime/types/array.py,sha256=b52d930967834cdb1d6aa8f121c29bab2ffc3424fc9aff132099ea931008b47b,13353
specflux/runtime/types/mesh.py,sha256=b3c19d0a5030860dc036903ba1bff4848603b143b9119c3707a96652bce052cd,8829
specflux/runtime/types/stage.py,sha256=869c01123d3e62c5d69bbcbed0643de6d346509eba8b4b7db7a0097b56d2f282,1335
specflux/sharding/__init__.py,sha256=8a6252d701b6a4106c9d862b8e165880d0875d52513abfa2d1b05b66aedc797b,1620
specflux/sharding/logical.py,sha256=b5f22bb6c19d7fc09e52f8bb2210631bb7f8c0f6f5b0587eaaf440810f305260,1850
specflux/sharding/mesh.py,sha256=d7445e565e628b8c993d633306e06d51764e504a69d2801fc3c61c12f396e318,26585
specflux/sharding/partition.py,sha256=457d06a501c0505933a824fa57de1d0a72c67b8e761a2e89cfdfa1620b7827ae,6846
specflux/transforms/__init__.py,sha256=fce7564bc2256d56006d66193f6c8e4bde505942765d062c483dad0b28cd216d,1532
specflux/transforms/control_flow.py,sha256=2aaa63ce1a35a72af1fb047b2ff41e305ca2acb9fcd2a220d38e1f7cc0753639,11308
specflux/transforms/grad.py,sha256=c058ac1729faa5d9d3c46710252a4e7cb29398abefc6d173d7f71197c99cc6b7,6249
specflux/transforms/jit.py,sha256=90c3378994ab220566e7f7a34be2a1490c42353d5354affa841135007029a4fd,9847
specflux/transforms/remat.py,sha256=eff977a9e5ab3c28f7fee6e8fa64fb0b16bd79e26cf5dbbde3efd2a64968d870,5860
specflux/transforms/rng_axes.py,sha256=f58c271d6d6fdeced0f455f0d47d0d4b580e593861955866c29cbe51eefa0a8e,4480
specflux/transforms/scan.py,sha256=06a3bc12b443d81ffbafa64a9d84a68f6378b5ab4882a28418a7bb0c6fecc5c7,4074
specflux/transforms/split_merge.py,sha256=e450f1e6a39f9df4d637d83be37b0a207c172822892bd74119cbcdbeea86834b,19758
specflux/transforms/vmap.py,sha256=7e66a549d4f4517dd3585474bf9922a59212e4267772c16f24898d1ed2cbca22,3198
specflux/typing.py,sha256=3960244a40eee5f010cb61b15d0282cdbbb788f1667f9b1138152349a5ecee0b,1387
specflux-0.0.1.dist-info/WHEEL,sha256=ab6157bc637547491fb4567cd7ddf26b04d63382916ca16c29a5c8e94c9c9ef7,79
specflux-0.0.1.dist-info/METADATA,sha256=f6a14e89db9b7f61bda1065c79619ca8b463224e82f0693c02724bbcb2a3473a,15902
specflux-0.0.1.dist-info/RECORD,,
