datarax/__init__.py,sha256=O9Kl8QXYQiRPOHtqnEjvI8PprmyrlWQbwzdxOPtw4Dw,2016
datarax/py.typed,sha256=tDlBv0KKT-EYYe71kGVWIzw9FkEX9EhBJJYveZhPgGM,97
datarax/typing.py,sha256=9bHXjoW-ISr2JDjvdGFJBW02Q66y53RczMWGbody9Og,3197
datarax/batching/__init__.py,sha256=MZ2AYpSgJmakK8AblrA3Iuw6x0niXcWCjKXbWMCPJj0,263
datarax/batching/default_batcher.py,sha256=P1uGcXlzyAM_3rRS9c_Chlph4QBO5D41llko-XHmTHI,6165
datarax/checkpoint/__init__.py,sha256=-2_4YfTD9FV7pG1U4GXLP4E0D51XfIAA3uXeUOZ5VA4,470
datarax/checkpoint/handlers.py,sha256=iL7v-6XBXzLK60DZ85ehH0695Ob-gwHnSEA3492pFiY,14555
datarax/checkpoint/iterators.py,sha256=91FP12rOKXrlnVr6tIcrMPigYgiHd0sqTL4Qkh4VCgs,6780
datarax/cli/__init__.py,sha256=RvECKv5zGF9ORMaHBzNH1DLxeqLxPK19GaQmmaal0lM,145
datarax/cli/benchmark.py,sha256=6KQ_puET_K_svs4jF02KFEG0zU89IFsMegyK6yQD-6U,6709
datarax/cli/main.py,sha256=Euw_-doYcV6zSAm6sLh43wJqZ51FtufSvrW73xQTUus,14459
datarax/config/__init__.py,sha256=yc8yoHU4k6gbP0NS2iI_dRq94R0KdFaoDs8RHMxAAsw,989
datarax/config/environment.py,sha256=ai4B4_AGprAEBLNLXcDbbRmfRChiNi6o0JeRBLtG-Js,4081
datarax/config/loaders.py,sha256=fC2cFqrA_kqg0LHoeX3bCEn1QpdKwwZt_FWI5ezYIAw,5125
datarax/config/registry.py,sha256=pd09qbPEh69yInkohx3bkw3z_Jt2yeykfXl4Pji0F04,12177
datarax/config/schema.py,sha256=7RIpjPpHVZSgKcQU2Ap0n1NDCNJpI3zcHLezM0wJJDs,9557
datarax/control/__init__.py,sha256=R6gfWKP-8s0dTB94wedGmgCYMU7jCNsbtP8pzTrkN7Y,40
datarax/control/prefetcher.py,sha256=7hjAmeZ2ftO84rYttfWxPP7BNRCFgorR47I27O1CQYE,12816
datarax/core/__init__.py,sha256=UVvemEm5wpB4fRMqP8Vr41quyzRa9Iew_FVe30nkiyI,1248
datarax/core/batcher.py,sha256=_5Au3JwG1mzizFCVi-3sZmrQL9b_c1U_HjHFx7yzfyQ,3231
datarax/core/config.py,sha256=H9bM0ogDcVfzciNt2MDqxcN7-i-FTrLx7grIPMfr8Qw,11666
datarax/core/cross_modal.py,sha256=RYIjBGW4k8ZXr87dBq6qtziN71wyVmh4ix1ablYAmTA,15841
datarax/core/data_source.py,sha256=7Wh_ft9vOGECbqK6o3mF4Gdp-6CRjevw9Uml5v8YniI,3642
datarax/core/element_batch.py,sha256=0OA9Pr2Lk60QJ012-t8UvbwcycEYAORqqBfSr-qryAE,26816
datarax/core/metadata.py,sha256=3kZNwZzShhfIAFRZybboP39suXR-_7tn0R5N70Mt9n8,16722
datarax/core/modality.py,sha256=QXGuM1yf6aOygAY2HAjl72hvd3mVd9asiijLXDBowAw,18342
datarax/core/module.py,sha256=Vgi2iHQhkWd3VpenNvIeZYdBMMuoX9DNjc2wrQhmZ8k,25543
datarax/core/operator.py,sha256=clepGffEIfuT577JW9AMSu2Lw0DPCkYqg5kTARn5X2U,16573
datarax/core/sampler.py,sha256=3ew_Gmj64S7z1NkOVx03i93debFFU0oELJu1pM76xfo,9841
datarax/core/sharder.py,sha256=si3NEFrmYAHHCgTHTtbqfNFl4w4WwyVor66B-QEB5aM,6708
datarax/core/structural.py,sha256=WVo3vXi2AU49xeVSID5lHTnXdiXw4WuGqXHFe9MJ-ew,6526
datarax/dag/__init__.py,sha256=mrQBa5V6nuz7LSTCqDGAsmblKbENgKOjeDNmHpmL4D0,956
datarax/dag/dag_config.py,sha256=k_bRGJoWT24rrPmhfMHxgxWBVgMPszP-wFIBMdvuX_g,6618
datarax/dag/dag_executor.py,sha256=WIUGuRzOUucIcLtaen4FNIMVugz_MuHrrYx-NvuW6hE,59404
datarax/dag/nodes/__init__.py,sha256=muJpDcmEBVvNNcP3lKLvoS-djCOez6WIMcsNY6W8a5c,1545
datarax/dag/nodes/base.py,sha256=dZfpUoaLGs3c-aN_Y8HQTmFKrYQgO6isWOr-8w9-edg,4621
datarax/dag/nodes/caching.py,sha256=xqoduJb6dZYEr53BSQhZM_Bfsy4x_aeIOAdTIoCSRfc,7977
datarax/dag/nodes/control_flow.py,sha256=FZ7mPrED8bC37Ma1Aa8WkAR7MrgTrm9ckU1ms1EYvRI,14287
datarax/dag/nodes/data_source.py,sha256=cAbbDA9rR1i8V-NwdX-U_tSISpFyiWFqhruPJFCspDw,20710
datarax/dag/nodes/field_operators.py,sha256=37V0BA_UwL9-VbeXCXxFsf6MjKXE7s0TWcGlUsd89v8,4913
datarax/dag/nodes/loaders.py,sha256=BNVoo7OtcgXzX9Jtpm5inl42FX4_CgytbsNB945lMdg,15533
datarax/dag/nodes/rebatch.py,sha256=rvhHRY_1vML9axR3p7gwvSzwA2rOGSEDKRzL3pLFuSY,22020
datarax/distributed/__init__.py,sha256=q8z8DBuy0GVV_r2qM5Z8J-Nl30yfHpYm--h_DUjO1XQ,2740
datarax/distributed/data_parallel.py,sha256=QoVkSDpya_b4uLHfRHd7SO-tQ71-PR1Aw0Nyv8dydyU,8462
datarax/distributed/device_mesh.py,sha256=KB6cscnjQo7drlbGhgoIvGpVb_R1qsPq_B54IcMMm4Q,5376
datarax/distributed/device_placement.py,sha256=SdXxQartaXhNH9u0QxugV3TN072nhAKMLLS5ve2od98,21447
datarax/distributed/metrics.py,sha256=ab8TJUiiRzixBK21runLhACSBUBgq6L3WZb_i3GfW70,6682
datarax/distributed/sharding.py,sha256=tHaqXwl4w07HNB7rcpaBkwvpAEqLYzxGEjwu-Js0ff4,3740
datarax/memory/__init__.py,sha256=hXouumuiTtikAclPDcq_Y4HVdJeqh5mYKFAj6Q_76Es,63
datarax/memory/shared_memory_manager.py,sha256=yGhNVZlu42mKGUAs7wjZPRRjGRK02ve_H8HDIb2iTC4,4656
datarax/monitoring/__init__.py,sha256=gZ3afc_T9Zj0BgdvzYa4G43z-9787zPFAsSJ4CvGfsw,663
datarax/monitoring/callbacks.py,sha256=a38i8iTKd0bYJuz_XT4hULBqzU5f-2_jBD_g4brv7q8,2153
datarax/monitoring/dag_monitor.py,sha256=AnuB5uzHAHd3zMcst4oD04DXxUgHyqz_FkpWS7l-5ng,21109
datarax/monitoring/metrics.py,sha256=dI_EQKVZpCgReDF4x95chvei-Q8ugjAYGasEXgX8s6I,7694
datarax/monitoring/pipeline.py,sha256=KpiF50WA8s5KRXcKlNCgcMqyeukL7eUe2zUhY24ie6g,8194
datarax/monitoring/reporters.py,sha256=fF-x0cz435xM5RF3t1KgW603nLAjJjVGnUkRvDLk4OU,8205
datarax/operators/__init__.py,sha256=vMKzuE3QI85NwVByZZpPa9xLGv2wE-kp2Bq-1Iit6YQ,1826
datarax/operators/_random_params.py,sha256=o0xgq5xgZdGMs2wXUG_jBCikDfZ4KepttGi_vC2hr9c,377
datarax/operators/batch_mix_operator.py,sha256=VqWiia0JTzGh8153g5oU1Dcz0_9d0Dhh_8Rb8AQ_woE,10376
datarax/operators/composite_operator.py,sha256=iuK6WyrZml46pR5uXPS0bivssq9rVM7g2fyGxscxYTI,20579
datarax/operators/element_operator.py,sha256=4FffpvqRz8nKoKBeMfax3S7J7aujguMUo_8Smq55nuM,6808
datarax/operators/map_operator.py,sha256=uXhKr7us1ALedi3aClW7zAe5oBMlAgIWbx1HgMu-qdI,9762
datarax/operators/probabilistic_operator.py,sha256=nHTPllfoN2A83Tf9WvpteUKCGiEPM7rVY4zliNARf_8,9466
datarax/operators/selector_operator.py,sha256=WzMZb5MPwGoSNyJZLPCaplqZP57Q-gkCtEaC7wcAmBU,9746
datarax/operators/modality/__init__.py,sha256=18VhynifeA8BTTlyxq1aRE7djrtXZA-arGJiD4etyQ8,1073
datarax/operators/modality/audio/__init__.py,sha256=Y-HJycWIDMCxLfQJBSApM4o8CgQF1OCewGxy-RWDbBI,1397
datarax/operators/modality/audio/crepe_model.py,sha256=ZzFL8-ATYVLs2xjYSkMmPFQIzIoA0AI0lGdRJQoTq4M,14773
datarax/operators/modality/audio/f0_operator.py,sha256=K-_N2HHinh1EuxghhQZbY68iM-zCGYZkxg_jkejDnXc,8210
datarax/operators/modality/audio/loudness_operator.py,sha256=VEhp3Yva-6ADxQZrSTY-fmnwBTMpDnxbIUQBJ9Y08t4,6599
datarax/operators/modality/image/__init__.py,sha256=Cp_gSlDVkN-RofzXwuZMNMaFSKGboDewUZiiG_QuZN4,1355
datarax/operators/modality/image/_validation.py,sha256=WGu_1RUtxqWIxaPLJRu07a-bmqHdjePAuT7kTBGqSJs,786
datarax/operators/modality/image/brightness_operator.py,sha256=1vO1Q7TVp3ogd-WvzOIMlQOOpW6vsgD6yA3CnYoEG6s,8152
datarax/operators/modality/image/contrast_operator.py,sha256=N05TKW6jlxlnwH1JTX1t5JMUag_prRAdU2LAZwJLiPo,5855
datarax/operators/modality/image/dropout_operator.py,sha256=G2gNfbT7Vr0kVndacZCdYdUmQa3TQ6I9RrP4eRDoAnA,11504
datarax/operators/modality/image/functional.py,sha256=6YulT4enZtL6tc01RzJqCTG4OSWiaE39r7gacDjH4zs,27106
datarax/operators/modality/image/noise_operator.py,sha256=_I6CBT4mNTIMT4e7E7DE-NE0C4s__kGEu_ySmR0NKM4,16166
datarax/operators/modality/image/patch_dropout_operator.py,sha256=t-lyR0NbetJNpOwmY1Oe8JaBy_QjqhrsKsUGDpQpbgs,12407
datarax/operators/modality/image/rotation_operator.py,sha256=rKQv9t11TruuPqT6jWK7rP3B8SydhNPHxEnFovDKiJ0,6586
datarax/operators/modality/text/__init__.py,sha256=9IHHyPqfbhTocHlo0DIS_0GPvv6j6g_TLbg3PnW9yEY,75
datarax/operators/strategies/__init__.py,sha256=uGYsHsqDClHukRPL7FcAstgaV7MA--fkqFzWcFWLAys,789
datarax/operators/strategies/base.py,sha256=QdUrwR9OHYjFKDJFiIXumSuYxI_R5CJLM7BkGJalb5c,4109
datarax/operators/strategies/branching.py,sha256=06IS-9-S1iFzRqH3eAM6MVNuChMpU0vQw77x022WuG4,4178
datarax/operators/strategies/ensemble.py,sha256=46zgz3QpWE4-gecYrNWftkbB28jLBPZYddkOC86NrvY,2114
datarax/operators/strategies/merging.py,sha256=pKBscrpPt96INKQyWqGFtR_0xSe1FJHHzRwVX6mwKaA,4967
datarax/operators/strategies/parallel.py,sha256=ouVm6iPrYQ4KmO9bAwKQFhh7xaN_se3mVNZxofbRRNc,7468
datarax/operators/strategies/sequential.py,sha256=7FH-G9qqUncrrsCD4QmbWOPETyP5pS3beajsbdC5w48,3739
datarax/performance/__init__.py,sha256=urhjnD1S9nMJ2Rvpw8hzftueyHkpHK6LUgVX-BAD8tU,983
datarax/performance/goodput.py,sha256=Uv8i0IHFJZmosxwlBUK6VIzi45NWEcZhQEtw7k-IFLE,5758
datarax/performance/roofline.py,sha256=DXx0NBE1xxb9fes0mVw7PVGxpUugVd6KCLrOOvMlROA,15778
datarax/performance/synchronization.py,sha256=zcrPi8VB2zyYuStCtBSGSCL18GPbPrU3H1OglMiB5VQ,486
datarax/performance/xla_optimization.py,sha256=eC7YYVnzSrt0YFh1ZUFH-vIsUSzhkHrwetF_kokpQuA,18852
datarax/samplers/__init__.py,sha256=Zkux1bTlRfkQNwEZTN88bLi_4VC7axz6SERWx5yb8I4,827
datarax/samplers/_iteration.py,sha256=smE4oxvs7fWQ6VB3A9l8-UqHNt5-4h4d18jnls0NXxQ,2787
datarax/samplers/_validation.py,sha256=UxMBmSsxiW8XpLcrP_mfqLf6FAZ22kTkxwjk3cdRQTY,582
datarax/samplers/epoch_aware_sampler.py,sha256=HeyWrPU0FBCP-qIMEbzwPsNBwL8zCuebQTsq9Ap-QxA,6452
datarax/samplers/index_shuffle.py,sha256=XLPqPZ64PT0pK4JXtWhhjZDrGHoylJDUJx4-WAjGknA,889
datarax/samplers/range_sampler.py,sha256=Caw1F1m6J-m2ZK3ZkKcxH6hACBF89SF9jJR-261iD_c,8576
datarax/samplers/sequential_sampler.py,sha256=LWCdhzUl7KWrTR2_7RtJzuSuhZlvtvSkNFrQesxg8jo,4670
datarax/samplers/shuffle_sampler.py,sha256=Zu1hQWGQJur7WvoCx3eMILucWdcYg-_ySqWbrglLIvc,4172
datarax/sharding/__init__.py,sha256=rmmzHxM7yQIPXjXrhA3lIyYobdZ-Wx5e_-HZPtYfgNI,327
datarax/sharding/array_sharder.py,sha256=DvqMTFPnzvpPAeh5enf4_nP6djYMSTeS7Oj7JIRhvCU,8612
datarax/sharding/jax_process_sharder.py,sha256=Sfm49ymo5jUSBl2TY2gteWB1gC6-jhZgMAq22SkYeTc,2959
datarax/sources/__init__.py,sha256=HwmeiGFfGMYZiN-nW_YtLPk8lbl9mbT1DtgtLi15IgI,11833
datarax/sources/_config_base.py,sha256=r6H-9yqRPVU687l8N2h2wJh-ADm2Ipr9HFX5mPPtKlY,508
datarax/sources/_conversion.py,sha256=MoSZmaHOW85k14B0ssaJfW_vkQxhqCx8OYEYqMNvvdE,6487
datarax/sources/_eager_source_ops.py,sha256=bQcLfsIhng7JPOp9lpaJUIvPMftOrMvIB8ghYv03NTw,16580
datarax/sources/_grain_bridge.py,sha256=wM7Y2mJFaYP-0aJ2GaDNaq7JS80Mx2lQpYfqdHXQoPQ,4358
datarax/sources/_grain_streaming.py,sha256=nK9GoIxaUZYx4jpKitSLGr7JMGOe7cYcrpfhmc8OEOY,3159
datarax/sources/_source_base.py,sha256=zGQQzMBP8tfItzEgNXUrij8BLAPb48v4_s_FCnjMf14,6501
datarax/sources/array_record_source.py,sha256=uYeu6ZVjT5F1TgOP4Ucv5yDDOpXHo673lmsKqCuUk4c,8352
datarax/sources/hf_source.py,sha256=u69K_T5BMwZbOMIo3CQI5kvIO3aTTWb9W7GPDEPkFdA,18489
datarax/sources/memory_source.py,sha256=zc_NVrirzGvdThJHyad7kmoADwsWtqx9JMeQ6Mg2dq8,21593
datarax/sources/mixed_source.py,sha256=bcOozfUKZqX6Ep7H6MQ54-FSKLgZdZowth5debfOvJw,5334
datarax/sources/tfds_source.py,sha256=ze6n_vVqRz2uYU79gY0QBOF7tFqR7dpiV5Zo2sdpvJo,19445
datarax/utils/__init__.py,sha256=YabK2mNdYKqrNj9IbKUADg-b5o_uLXNgHbsRzs4WO-w,321
datarax/utils/console.py,sha256=rbStwPFNxlC8QHZ2_BlzAtvhsPtDNu_oLKFZ8MNP5D4,505
datarax/utils/external.py,sha256=IMYVmulPA4gu5AdFw2Y5emm-strLzBmD7JI90uEVJCY,10462
datarax/utils/prng.py,sha256=6x9OogTu4lgdvkYRRjtIpwJi3IJajuzyUN4YMBFrbnA,1680
datarax/utils/pytree_utils.py,sha256=xbhTjUDTCkIVjG3Kf1bEqjDnxFU_6aRi2_Zj1A9fBSk,9257
datarax/utils/state.py,sha256=YXEA9KrTg95kvJRIFMF57rCzmHdMryKgoaQ22o3Jf4o,1715
datarax/workers/__init__.py,sha256=ZllO8VCSBuJSZIyfVQ5F09pQEpJG9nZ8lWncYbbRxaI,55
datarax-0.1.2.dist-info/METADATA,sha256=SFyCVE6xi1Y13nmj2R6HiSUc44RbfpcAr_m1bHmT_kk,31050
datarax-0.1.2.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
datarax-0.1.2.dist-info/entry_points.txt,sha256=K1-BS2Gw_6J3L82rUied-ly9CR1iM5lw9Lux0xE3c7w,45
datarax-0.1.2.dist-info/licenses/LICENSE,sha256=RbDEoCPxPzEkZeUhMGe79gANcuvNMKefRDs1FiHQlkY,1070
datarax-0.1.2.dist-info/RECORD,,
