reax/__init__.py,sha256=WB3MouY8QkRIOKro96qMgP0kaRGkD4moYbTiltsGmZU,1348
reax/_engine.py,sha256=q1IaFx_T1FJdl8009z98tbug-ckLOiVCM-hDNoTHzyQ,17836
reax/_module_hooks.py,sha256=DkhSGQdZ7nMpKcbmaZpoMLrmsAIynZcEQYMGrUewGE0,10256
reax/exceptions.py,sha256=9BfWSQNrOzGQw_Dr3ARs-tvqcIIkQRRpAoEExtu87pA,303
reax/hooks.py,sha256=kH9Ob9p8ZA6RSxiNWLFgYu5EaYteEbVRCLUF3LzvI2M,8951
reax/keys.py,sha256=fkWzDL1R3QrmeOAzEQDRvENu9e0JP31UWwvbIcPLLR4,292
reax/modules.py,sha256=obC24S1Imv6Hy9lTVZaZo79tT16jIrYYArAXK_ucrBg,11364
reax/optimizers.py,sha256=jLfSZZhpVYsfTofbFyvcx0XfGGoxKn3oMFcckfjPYz0,4783
reax/plugins.py,sha256=OmL-6jAUE9RASV2bpVtVyXdmVwmEME9widsAO740NyA,1714
reax/random.py,sha256=5KUMkFqiA32RbhhAj124RjxvEgC_I_qDNTHiv_nqKSI,383
reax/results.py,sha256=WFDyiqJFSYy7pGEUryFWKQlfsJP2UJ5xUK93v1z7UHk,6076
reax/saving.py,sha256=ITjmzM3jRJQs0LGxphbHwUeiG0KVGMp7H7DNUbDHb3c,3628
reax/testing.py,sha256=BZGGeTp47e2DYk1R5Htik8bGC2oQ6sFxGbs24biS8no,7251
reax/types.py,sha256=jrYS5jrUxcNqGYw6x5TD0giuq7DPR7ARhJxwSFdBlBE,1786
reax/data/__init__.py,sha256=7LrLFS-jjkjSh-kV02fimGlAeJ2Df_kIxtk3R-u0DRk,748
reax/data/_datasource_manager.py,sha256=kUsCFO8K9BU4dYr7pVuzkQl7SRKUkRvbCakLVhpkeQM,6767
reax/data/_datasources.py,sha256=5DXhB78o3p_6wC5915SUDROAMwzIEWq6cXaUxsELkYo,8783
reax/data/_loaders.py,sha256=ExXPDVwnKGzL_KhN_1dxZjxKy8Towz-PeK-P7nEKDuc,7433
reax/data/_types.py,sha256=61LZ_srs7Y7P7B-jc963qtL2A1CQG9d-z0xJNAEsY9Y,1206
reax/data/collate.py,sha256=3HYf_Zhfz94BP1PsbSF_mGYWa-OB-c2gG0gEWidRET8,5587
reax/data/dataloaders.py,sha256=WWEYhmpRG_85eFUCwv-G2z9XOB_NVO8JG-1hsobhV8E,1096
reax/data/datamodules.py,sha256=r61bnekuWj8KUicJzHjL5-LiBc85QqssD8Y0802edGY,3065
reax/data/datasets.py,sha256=rBHxx3XOuiO7cVcu3MEGq-1-5Mie6XuOo29qqyCNKAc,6993
reax/data/fetchers.py,sha256=VM4vV93R9Ym0rIUe7HHwAxbFaOZjmJtJI_pJc6h-qzA,4120
reax/data/kfold.py,sha256=jTududrFpZg91S38uqVNO3NEXjaFJ82eS9GtWag8i2s,3156
reax/data/samplers.py,sha256=COeEdEvBqzjs-q-CYkr5gMWf635erKupc6L7pVGbTOU,12340
reax/data/utils.py,sha256=72qWmy3iZnrMBZu_btG4RsyX_DFVdIpkH6LAZAX1hMM,3639
reax/demos/__init__.py,sha256=QMi9FV6XNgTidUlhThRyHsuu85fLIF2iLxkCGjy4XTM,113
reax/demos/boring_classes.py,sha256=sINj-rn1cSh1bA_iERoxrvvnZCcySMfeqabD8G0NfFQ,5700
reax/demos/mnist.py,sha256=cRGIB93WWJGaEjLVkfpKLNvgfgxAbZGmpSoW3mb0d1Q,13240
reax/lightning/LICENSE,sha256=pB4vrpkVrlYCj429C_J5lfGQdhPLzS2BphsBCtNOn-k,11349
reax/lightning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
reax/lightning/rank_zero.py,sha256=1ZxdAffiVvnmjRXDafMLCyM0_uVtTQss_E0qWdHX5zY,3726
reax/listeners/__init__.py,sha256=0_61VG-QcHuYsCjZN3wiUBShpS4O0Hn8FaDgX9-UNrc,373
reax/listeners/checkpointer.py,sha256=GHAyYlXXmTjfoA23ZzE0FsRithEEdJWw-ffjQL5fi14,145
reax/listeners/early_stopping.py,sha256=UoCc0qqpyG0f1uOmjrQJsXeMGDLEBIQnlwKp5agtYyk,12980
reax/listeners/model_checkpoint.py,sha256=DWxrqlFNQ7QbgaC5Hwf0THNilcyfClXYUnTeobZ16pQ,36744
reax/listeners/utils.py,sha256=cDECyhGgoBPUQrPo_VhR17B_3-W0-FE2REAeqxUVqhk,285
reax/listeners/progress/__init__.py,sha256=z6J45f-50XpKNJHtmFC2Ion8mW1FQuUWE92LyHXvFUI,244
reax/listeners/progress/progress_bar.py,sha256=vxdAx0cIvVo7QDbNEzF0mRSXENyUEe5R1gq17p62ZZw,6738
reax/listeners/progress/rich_progress.py,sha256=bhtlAtClg3Wf7hVPZEkBEABB8CFlf_eq4UcLSn0DUzY,25849
reax/listeners/progress/tqdm_progress.py,sha256=hRJoUAqpE568MXy22gMr0UTWTNWdBQPjbtn1qwyPyZY,12892
reax/loggers/__init__.py,sha256=Pza36kXNtyFX-JmmW2vQxh8lLAMlvUx4YHQgJN5irbA,231
reax/loggers/_utils.py,sha256=hexaOA66BiplzKScySvNbSHRF_XRJui8U4HMoj5f0qk,4587
reax/loggers/csv_logs.py,sha256=zci2xVV8YZx6S52bQ31srSgCrEcy7krqIDuEgnyg0W0,11144
reax/loggers/logger.py,sha256=inTh7UbsR1xfJK_PtbXA8oF4suXYCSVhUYWH8bQY8Rs,9506
reax/loggers/mlflow.py,sha256=Du0_D3KOYaDc9L_aStIVulsOT8Qk08G994Mi_ioI2wE,17813
reax/loggers/pandas.py,sha256=8A43YUiTopOK3P2sJBb9Ic8_tdfTn9WMDn32Dyg0UGw,8229
reax/loggers/tensorboard.py,sha256=ua4ZGH5QuDAwVZcd5mpOeGqGdB8OwCCSZYgHh6YlGcI,13297
reax/metrics/__init__.py,sha256=rz-TIJVX8nY7jkZfqQ9MjgUl61obp7Nvclb7Xd60C7g,625
reax/metrics/_classification.py,sha256=d1nnjNvZVSwAxwi-3rRzPy4R1LikJjGOEsm8jkK-Imw,13752
reax/metrics/_evaluators.py,sha256=iCNfPhWV12w_eywi7S2xNslw1fMOvE59t855pryCIpU,1001
reax/metrics/_globals.py,sha256=uNuV8i0K9Ibb-aiUwynGMFdtRCX56gJab09Z6dXGH0w,621
reax/metrics/_metric.py,sha256=iXTd7QFQ8pYROaPBwcYKYGR4B_SvprG6UkqdO3WnqPs,7640
reax/metrics/_registry.py,sha256=nmN_E8m84eq5Rj7Y-IwagMdexMSOmAcRJZq7lPP5jmY,3209
reax/metrics/aggregation.py,sha256=LlZnghRM_CG3OcrEY8-YU3pJh9HXBuGUEbZTpVAlkyk,10677
reax/metrics/collections.py,sha256=xIho7Va63ZgpvE3_fU5a7bxtjsV0z6qxoact3mshml4,4166
reax/metrics/regression.py,sha256=gBMeKr6P5H6vcn4ysSyAjAKy6vHM7Mv42jZUsO-owJo,4118
reax/metrics/utils.py,sha256=ueQCsYTWUcJGhiCRi0fNChQBTHP8xTeQMd5y-VVTRxg,8372
reax/metrics/jm/__init__.py,sha256=eKSafyoeL6KMVu8p6q5Rt6ciEVqZtzbRn18osDw2iXg,42
reax/metrics/jm/types.py,sha256=KYcKr7BMsxaChtISrIljqqCk_kMQYBmCpuGx9-cmf-I,1446
reax/metrics/jm/utils.py,sha256=hv0GwuxHboTxq_mQRPX3ZS94r8X8kCyfEZlXvoX2idM,35642
reax/profilers/__init__.py,sha256=ldDnzhJ7nVxh7ef-86_iSeWNINaoLWUWrh6yRSCPtpY,140
reax/profilers/jax_profiler.py,sha256=Vhvputu4KIj0Vlny0puFmySZqFaLyQ5YKI2V7_8Abx0,1896
reax/profilers/profiler.py,sha256=C-KRsQU2SIzaTc2fUQ8x4tAH1iOrPLSUyAW0Km4qxmk,2166
reax/stages/__init__.py,sha256=hGDRmj8tqicqrSW6ignjUOD5-h1YPAwuqioab3nAylI,530
reax/stages/_loops.py,sha256=UeW2lkXxELU2bhfZVMBPaXM7QPbRAeq6QtttsnoMpKY,19763
reax/stages/_timer.py,sha256=ttmO0QxbnsaFKasH3a1BSS_FXvbXoy5SnhwJPe2UCs0,2746
reax/stages/common.py,sha256=bcDZs0v21mLVVhyRSecg5-k3r01IvRkYoIE4z6y7zQ8,8920
reax/stages/fit.py,sha256=po51zyVM6QZCPG-og6xX5JoraPY0P8FUBX7SlUc7fTw,16224
reax/stages/predict.py,sha256=yLojnI_KkeEjVIbPTN7GuzAjlYJuYl4pxK1D56W7n-I,4114
reax/stages/stages.py,sha256=v9EOL0_eREJaWYuJoH779oW-06b1rdnXQJFYHCjHii8,10791
reax/stages/stats.py,sha256=AfpXqsbYAmBrHIVRN65Iq5QChovignMjHZq88gBUtxE,3090
reax/stages/test.py,sha256=hVZsgacXLjHgXbxnwYvT7AJ61H9DwlEAzdRTs7SfpSk,3314
reax/stages/train.py,sha256=jURcdmmFFkMAVPA8NcYoPLybioxKsvluPrhGjyU04SE,16503
reax/stages/validation.py,sha256=IiilQ6-eP-wHAM9Ydq95hohpeSCC4ERP-EzIQRfvoJE,4461
reax/strategies/__init__.py,sha256=5y0mm7wVvZ6inmVPooKIwer7JwSIHD12jmMCE5yq0Ak,329
reax/strategies/_factory.py,sha256=6i_FYP_sHWbyOl6W1CGcu7FLqxBXhDpQ1gl4H6alWs0,731
reax/strategies/_jax.py,sha256=_TIEscjSYD8Pu5giahow-9g-kXjQUMg0lS5Ns7vjir0,11901
reax/strategies/_parallel.py,sha256=_8LQzIJ4n6JOSc6hV8NrI_ejR0eafYSClLrVzLIomHM,2248
reax/strategies/_single_device.py,sha256=lMmc1DkGQOf31X06h3WX8sZT_5NrmYCO53LZF1TeV0g,2107
reax/strategies/_strategies.py,sha256=dmSAfVg2kCqxRi-EkQvt_mlIwHa1cSksgjxe19CpS1M,4427
reax/strategies/_utils.py,sha256=YMxK2iT5bHBnsBubkj5stLtSFNGlEtrZKuiuOva6fAM,692
reax/training/__init__.py,sha256=QfRGDaX8eYODgnFpCAVIYU8-dw58sup0jCkQh36xsKM,143
reax/training/_checkpoint_connector.py,sha256=Czu-Ya8KFKn8HwJcPLUWjjpRKUKgY6-hDOGNW1glArI,888
reax/training/_checkpointing.py,sha256=KIlRzeZ9WnHnCxSAX5PiMl45fx53iCxqEoprtx8FrmY,2188
reax/training/_hooks_map.py,sha256=heODtRdhMXBlL5PU1V_NaSuXRC5H2MPcQ_y2jqGpbAw,5135
reax/training/_logger_connector.py,sha256=ziFqwf_dn23bnRQ0bE5W9SyhLKVegTbXK7-r1dsV8M0,1968
reax/training/_utils.py,sha256=iGGE-GJKiBXNGvh3AzCfIe_hQQuRyisr0Y_HR03Ee4U,2683
reax/training/trainer.py,sha256=9Jkx8Auww6EF7OedM9Kr60xZUDA1M0mLNbxUL-fCHmY,41319
reax/utils/__init__.py,sha256=TNIt41sn2ZFKnfTJc26dRkcp0ZHIBgLOV_iX_KXdJ2w,131
reax/utils/arrays.py,sha256=XWx7nNW93VKpM4LgBf_3jr4r39UdT9zj3rlTtrW6elo,1062
reax/utils/containers.py,sha256=0zzNpYZAaGhjwyKi77NOXp8i5AOlJ1JQbWc2fjhZy9U,1977
reax/utils/events.py,sha256=Q7RVSEkcJqjPwr6JmfiG-7KrlqA-7c91xcmT7mLFy2E,2147
reax/utils/grads.py,sha256=xnX2Se_NQAqITOiAJplWXD1O4-fM2Pf2CzO_o-TehWs,3330
reax-0.7.0.dev2.dist-info/entry_points.txt,sha256=G_PEmVlBPST4dmlJOITb45vbmAnBYEbEOFG2dkg87SA,302
reax-0.7.0.dev2.dist-info/licenses/LICENSE.txt,sha256=46mU2C5kSwOnkqkw9XQAJlhBL2JAf1_uCD8lVcXyMRg,7652
reax-0.7.0.dev2.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
reax-0.7.0.dev2.dist-info/METADATA,sha256=-HouBhNvu8xhmegjGeYvSEQ1-twgRwR9-YAszrL_dnA,8088
reax-0.7.0.dev2.dist-info/RECORD,,
