optax/CHANGELOG.md,sha256=IDfLyPu_gGvwslZh7-GeEDwQPad982b7C0TeRnZotzw,1383
optax/__init__.py,sha256=a3zXcggxhkPSmQAWX8WMyHQTELmk1SKcly-m3c_2tEU,18197
optax/optax_test.py,sha256=T7MA3uQr8f74509AAtvXAdcxpj9-GeSpr8yh8N0N6Vw,1079
optax/_src/alias.py,sha256=xadJGpVjj581fP6lS_iHuzZcU7qnY_nZnpQ0JFBpPLo,111735
optax/_src/alias_test.py,sha256=vSUgtLMOxnNRttmZGfKdqdkEhG4FF9hfAwsv1ZeXNhQ,38302
optax/_src/base.py,sha256=A6RkZ-_dXznNiizUPzf4HhyxqMsaKnADxZrcw-azmZI,12503
optax/_src/base_test.py,sha256=6NgeqjeUE5P-WT6Jp9Z-1B5tMDLDC95JB7dpSOA1eIQ,6456
optax/_src/combine.py,sha256=wyD9MPOZMlB97A73Ns_JsEk9btsEfcSnsL9Cg1ZZps4,1000
optax/_src/combine_test.py,sha256=RXphhhKuMQVkdEWNQz0oWBQyDqVEHr1w0-U_hjZGn38,9285
optax/_src/constrain.py,sha256=tYrEjizwVIqtvkXZS9kLf_ZAtmaeqGhXH26dxfWzEDw,1016
optax/_src/deprecations.py,sha256=otkPgkovQBAPKytIkQct1KElZ3qkjYxMkxU_A1OkJ2o,3131
optax/_src/factorized.py,sha256=ethWA1bX3CSbrWbBrfnh4NWwVHdNpiOmdHbYWrYk7Lg,8257
optax/_src/factorized_test.py,sha256=Kfoj0ArCO9bT4TGXgdEfRwtkbhSXbtH2OmczeSNFolA,3473
optax/_src/float64_test.py,sha256=iJ5P003Pj-EZ40oYSyGabjAeNhG-CyZF6b-46Bf3SyQ,3780
optax/_src/linear_algebra.py,sha256=Sd6paLmZYlP-tC4qCVzAGOFklmsiKEW20OH-eKNZMV4,12180
optax/_src/linear_algebra_test.py,sha256=CU_ePtRhNnYdDrbUwDk0_rX2r4P82YmnQPmZH_vo9bw,8206
optax/_src/linesearch.py,sha256=_ul2GOzp-vTqoKOCfV6DIYpv2mhwN1jLNVFPkfsVmVE,70062
optax/_src/linesearch_test.py,sha256=yKbFnL7g9W9TisexJDymlilAoQTC4qmA6Id9lSbpPRE,28878
optax/_src/lookahead.py,sha256=TtI-2dksiG5oZ2FIYkn_WAI1e8vP6fi1kWTaa3nCm74,7650
optax/_src/lookahead_test.py,sha256=v7REaBww9MDX2CTSdzGtjrHKmjHWj6l283dyHl9Nkbg,5556
optax/_src/numerics.py,sha256=5SfcB_jcme5vRXks6FDVq9y7m5ghtInVVAE9nz57Lio,5487
optax/_src/numerics_test.py,sha256=84HziGA1p1HhyFAMtaKX9mFDJB0x9zs8OjhlXG4mHys,5734
optax/_src/schedule.py,sha256=-JrpIPu2wwLPDsi-oHejHSsCT_Qz9K9Idd9IfwLmVLk,1980
optax/_src/sharding_test.py,sha256=IgkQJCg6Y7EXRFmzESs9Bz1PpKmz2csa91_ENzGm0bE,5191
optax/_src/test_utils.py,sha256=5XfWJR9DQRFTSPel2jOLOeQB4MsLayvuEAsnJhAZNYk,4021
optax/_src/transform.py,sha256=StkuiacOK6RG88ZSFSATilf7vz_VST1lCVUFstA4mOU,63394
optax/_src/transform_test.py,sha256=Z-k5rZcOEpO2gA2rS_oZECghS2Dx3NpKWhCcjurggoE,9029
optax/_src/update.py,sha256=BzEIuHWtROJG1HKZrvnhENo4oLeT4T5cfWaGyFTGfdM,3891
optax/_src/update_test.py,sha256=L43tTOHOX8F_pJ_nDbAAYfC70p2SvMETtQzNYmThCLo,3185
optax/_src/utils.py,sha256=3JVyXSizyWlOPhmjvnXTtja9BGfPTuQjIQBlVB-rrj8,7362
optax/_src/utils_test.py,sha256=OIByOxzfiEJ4TSqo2HD0xn9vPOB6nh2bhLOXpkciSnU,6844
optax/_src/wrappers.py,sha256=EEq5W1v-l3TQXMSP4S5VEWCrTx93ROs8efFjKfz4020,1676
optax/assignment/__init__.py,sha256=u97sZb7EmBvQh3Oqw_XoEtF2dwZeVg5IhsZ0Ju2WeHA,913
optax/assignment/_hungarian_algorithm.py,sha256=mWZWQViQ-oAXo93xxrlcrKsJCxORNWOnPO2iGO-Jrb4,14438
optax/assignment/_hungarian_algorithm_test.py,sha256=y3CUUs36-38k4M-1Jb-sul-jN63hQAtVUCXDrWBlslc,3867
optax/contrib/__init__.py,sha256=z62naa6lGTMBvuNNKjLqJ-x9hBS8SRKOzJfOGHFlXio,3673
optax/contrib/_acprop.py,sha256=8jGG3uTVGln5Cxs_sHTghhJeXvoMV1CDwcC2nlMUlDg,7007
optax/contrib/_ademamix.py,sha256=LrznImSdY664-GBSEBm5VC2h-8_wVm1uJ91jh3uqlM8,18833
optax/contrib/_adopt.py,sha256=7I9S2kEfiawo2bATdqJ9aRfJWUvYDkSO_KqrNtPW0FI,6790
optax/contrib/_cocob.py,sha256=LFbvNCouNfFU8VKr7y_i8EOz1v8iIUZsSPYqJrzY9Mc,4575
optax/contrib/_common_test.py,sha256=zRkR9NqY-DhiX9bGxCw7KgU0tvKO4sCfy8MBfu2IBbU,21905
optax/contrib/_complex_valued.py,sha256=qyr7jwnmJ9lCaSYrZqKtKs7E8_2HQQbzXGrVujvW3_k,4301
optax/contrib/_complex_valued_test.py,sha256=cVUp7r4qm8DRrf9n8f2HAV5j3PZzroam0uP3gnr47MY,2885
optax/contrib/_dadapt_adamw.py,sha256=fmlwzIvcugZiqKHl4uG4oaL5eU9mRkPEAKJ4Uvuz3E8,5645
optax/contrib/_dog.py,sha256=XVK12JOX13unpWr4f7nmGy9bVP-sCubTttBea5G5c8k,12609
optax/contrib/_galore.py,sha256=8zR52HzXn8Qc8RUyG5fpoW3hzxzUgApbSm0Iqf9V2ME,20727
optax/contrib/_galore_test.py,sha256=eWalYiF9pwtalhuVISbxG5ONTVUtOyH1QSeC8ZD5jvc,16136
optax/contrib/_madgrad.py,sha256=_3UtY3UcXQxePSo2UmZojtECMGtN6N058IGhbNPeNU4,5409
optax/contrib/_mechanic.py,sha256=E4eDSUfDauM9lurPNx79R_AGQFfMriTRjzPlUeztooY,8107
optax/contrib/_mechanic_test.py,sha256=C3hl_uO4DCbvV4wmYlrvFmLzCIKqe_Nmyb5JyJeb5jY,3722
optax/contrib/_momo.py,sha256=S_u-6ck5BEX2blw9IbtrfuM1AYuR3-YnLUNBteu6Ciw,13143
optax/contrib/_muon.py,sha256=GA5P-xKRz9ZdSkQ8p_5fWMZJf3uanaRD2bLFXk8JhOI,22469
optax/contrib/_muon_test.py,sha256=O84_G5x88w7_qQpvitUeqO2zsWZ4FEBqQJccH9y6t0M,13751
optax/contrib/_privacy.py,sha256=pKvvdgECsKPVeSWEzdpapzTR-Q4FnqJm7eapdlPmTfs,5793
optax/contrib/_privacy_test.py,sha256=m620GsXvcZNGvUf0RzxWfDdlFtl2EpUhyjBMQZVccgU,4372
optax/contrib/_prodigy.py,sha256=5cXvVS8VFJ7HtHbxbt0dtwxfWtOSVx27dMdoV7ikccU,6557
optax/contrib/_reduce_on_plateau.py,sha256=k6Tbo5o7_FFDfSHSYomaRlsWGWzN6KN-cc8KHUF5VtI,7067
optax/contrib/_reduce_on_plateau_test.py,sha256=y6lwHM89C_3gbswErwr6B5vTuwZ_IZTXb3M64iXIWVU,6565
optax/contrib/_sam.py,sha256=lxRj6tkPAxWw01Zi5-pu86-P3J20PVWKx8T447mBves,9947
optax/contrib/_sam_test.py,sha256=BOgVFaX2jJUU9MumpjuftivJAx7QAiEIVJmEyKBcWHw,3275
optax/contrib/_schedule_free.py,sha256=x8YbSIu1DuSPXGHh-otoBYnKLB2_qxJnJCnKkv0iC1I,14366
optax/contrib/_schedule_free_test.py,sha256=yzpX3Axs7fNq3fCB1I8LRsDKXJ1eMIPzJGxdfWlE2jE,5417
optax/contrib/_sophia.py,sha256=KJiJvV97QnrcST7ZBMcxJLNUpnBzNSo6TGLtSr3q3HY,12354
optax/experimental/__init__.py,sha256=Fo22nhXr7Y7F7snkiu09J5Q5qbZzD_0VBYLaMVH9c-8,809
optax/experimental/_aggregating.py,sha256=j6o73bMy6VXX-yt1Yrso1mzpI4BK2SoawSufN1BCKSQ,22302
optax/experimental/_aggregating_test.py,sha256=ZrTnvEv8iArYEio--2qM0SYRTN0CYczym7yha9998I8,8288
optax/losses/__init__.py,sha256=AN1kSGzYouPY5RwRjMl2SMyENgz9uDAwH2jE41tVxsk,2743
optax/losses/_classification.py,sha256=enE7FqfXRasAT6M_6w7kIFCDGDv9xv-3LTMkjemRmSY,34732
optax/losses/_classification_test.py,sha256=7JhBrsozO-_LeAevRUrZHGGogSFw4o0PqlUMFjjYNqE,38486
optax/losses/_fenchel_young.py,sha256=eNhkJh1vwITDiAYkj99Kdr-j4QhWDvRAPuM9pjtDYnM,2113
optax/losses/_fenchel_young_test.py,sha256=Dr1GWzsoPzDOPoQlaBhWRdUy8duPKbTlxKc6uDZfFAA,2688
optax/losses/_ranking.py,sha256=17p5_42ACHjpWvMlvVO5Rfl4_nZ5r3gXeTRdjg2Qy1o,6335
optax/losses/_ranking_test.py,sha256=l5j98ik5-sVfbBa6GSlffmtnVb1dQ_XbLU1mxmWxs20,8941
optax/losses/_regression.py,sha256=PJp05BVmvw4wxSes2jsKfzQM3fk2CNTsZdhQywFN9YQ,7990
optax/losses/_regression_test.py,sha256=oHeIXN1D7ysHE0GlJZELTW6tVQehA228ddOIomew9Ac,6776
optax/losses/_segmentation.py,sha256=a-driLtRysoo980UUShAhthSWrGoUdqQbkLw9-DP0tM,10153
optax/losses/_segmentation_test.py,sha256=uh9sm-811mjiKAyHw8jtuIkG5d9Eq-tF9fPD3jDKnJg,20235
optax/losses/_self_supervised.py,sha256=2dHpX6iUnN4OWEq-hqnG3vLpbN8DCSlnMKcq5km_Npk,6408
optax/losses/_self_supervised_test.py,sha256=qAQxCu6fMFJjy-wrivIafjdfjtMG2dydI6dgW1B807I,4121
optax/losses/_smoothing.py,sha256=lGTZdWb0Y3HZR8fi7X53gLJ8gjHlEAZzwgg3iT08bTk,1882
optax/losses/_smoothing_test.py,sha256=WE90aVySJ0Wp5uFZmqjzCP2IHWHFAxBUGuMOtz-KFYw,2333
optax/microbatching/__init__.py,sha256=gsGyT0AVdRv9_bzR1sBNSPeZPftXoPVgyqD5pJIAEsk,1052
optax/microbatching/_microbatching.py,sha256=YIPlwTxnmYgX_Oy6tju0YljIatCTYTxjxIR93Y2KxBg,24836
optax/microbatching/_microbatching_test.py,sha256=92GS47kKX0udvEEhf-dW1QBsfRiKXpBwrukdPM5yHfI,12289
optax/perturbations/__init__.py,sha256=VMj_AXmy85zJS4FmyljYqwtv5pQdzBtT9eQ1IQmUrsM,934
optax/perturbations/_make_pert.py,sha256=e0TST18ul2TVKxTjjj3IMKjw61iGsgKiHRNpekpdhPE,6301
optax/perturbations/_make_pert_test.py,sha256=9fGzbwz02JfirBcp64Xzl2vCVtueewgAf5rbj2EPW-o,10244
optax/projections/__init__.py,sha256=NOIj2H3EW78oS-x3TrFbp3HsdulkOQBgwlF5vgANeww,1528
optax/projections/_projections.py,sha256=_o91FLStdAAuNFUczLPIXZnfBNvzcey4hW4WWUex4hY,9957
optax/projections/_projections_test.py,sha256=lw_a8J-9JbCYIIpidOHxdc_fONX7T4j1X8cqh7O5zZY,9722
optax/schedules/__init__.py,sha256=Co2sonQPd-qNHaeJieqCRZAqvKYXzGrmoDlGKh4-p0c,2002
optax/schedules/_inject.py,sha256=IfRHGKUGdIBLfd8T4HgLO7IlbZtSiIMnvRMxaWM5dBg,10471
optax/schedules/_inject_test.py,sha256=iN6hInjFd6iSnIIKDzR1TP2_CLEQTfF1LZV7dwEKLIU,10665
optax/schedules/_join.py,sha256=yhCsqCLCVYrfECxkowYeSa5cAbtlWrLkfz02az2GzJ0,1659
optax/schedules/_join_test.py,sha256=NFjwadA4Udwd1bDmeJMBwa0tnG8bmuDiQsidehD6zLQ,1654
optax/schedules/_schedule.py,sha256=H-oXc6hotXoVUUkT0hcPe6JZXjJlYszYuDUK_tZdras,26164
optax/schedules/_schedule_test.py,sha256=Fk0h7kNhiYNOFsvS0Woblr75LGOrwL5rd2BxR0Jfpd8,25143
optax/schedules/inject.py,sha256=r-Zzd_oyV19qOr4m4jfePuI5y8Ih72_tpjo1zjyQ6Xc,897
optax/second_order/__init__.py,sha256=cV6e3xmxmfaashqkcMrQmi_qtFfbQqrxDF6gzL7SmdM,942
optax/second_order/_deprecated.py,sha256=SSBVbQqsuZkD7-7aImWAPIgmA8KFyiOyNrvPmRYTtJY,3774
optax/transforms/__init__.py,sha256=LMVHMuUKEr3BAdyaCz7OPy_oQD7-QlAQJyrrPmBBx7Y,4226
optax/transforms/_accumulation.py,sha256=TodCHoEAZEXzuEESjbxzZjCRkHDGnpDcCVX0cikpO3g,15636
optax/transforms/_accumulation_test.py,sha256=DZ5NWTtAh4prgP5yo7q_hJGDwmsPdQH6oIL1hBiRFsw,14346
optax/transforms/_adding.py,sha256=ainkBkfvPQKQe4mzTTgBlErTPP5nrplQPrkSHIbFS1E,5876
optax/transforms/_adding_test.py,sha256=JTJZuOZDZX6YgWUTJbP0hws0VR4ti8qkRIG0116tYpg,3721
optax/transforms/_clipping.py,sha256=I2k0x0AGkkWMptuJ_PzLpmdS64pi5lXxHaFEY5mU57Q,12888
optax/transforms/_clipping_test.py,sha256=moFg5HRJf3q3Q5PwjHngEO4kQkwwA_e7dbi_Bexq_4c,12446
optax/transforms/_combining.py,sha256=O5kuGpb8P8N9jlhfG2xIO--wSJZypanMdlOX53liFSo,11134
optax/transforms/_combining_test.py,sha256=UuTrRYOwiRHYPKra0QfLUWc_x2w8DZvzu1jKtD_0uK4,9325
optax/transforms/_conditionality.py,sha256=nlGX4Kv12QHy5LufVXap2vuZdi6WrRdMSOdroJrV5PU,9449
optax/transforms/_conditionality_test.py,sha256=rYKHPUmjkcxwm9a42RRaEDqKLJsrylB9IMbm47-juv8,10443
optax/transforms/_constraining.py,sha256=r7zzlSOJsu5Y907IxXN6K540m2FK9_p2ZvPIkeW51Io,3256
optax/transforms/_constraining_test.py,sha256=VYarjnJQWdOJczwYN5HB-GoxVi8x2n_TbyVr-PrB7vk,4669
optax/transforms/_freezing.py,sha256=QNjzhN4BGR1_HBQJvIXIl224f7wbFfM2ckvLkQfkvFY,3846
optax/transforms/_freezing_test.py,sha256=k52LIC21jdZsNIiOX_P4cT26OKLSrg9NizyXqD8EW4U,11912
optax/transforms/_layouts.py,sha256=fk-z2A_BW1rGt8GsnwqQ3t-4GKsmnOqcKenNvLhsfCg,2616
optax/transforms/_layouts_test.py,sha256=4p5SvW2JVnIQVauK8lIiBWSJHscmN4_j1-CbCQP86m8,2048
optax/transforms/_masking.py,sha256=mqYJHB99tJtYtn8MczNejKevdfInH6wjE6rxzFr3XSE,6262
optax/transforms/_masking_test.py,sha256=UqVYcpPDC0QPweEpSFnRrL5hxr0C2CKiYOhnjZvtRyM,13221
optax/transforms/_monitoring.py,sha256=PKeKSHTsjvHdwiLYwfFxb9FUV-0meQFCmsU2N7hYVhA,7465
optax/transforms/_monitoring_test.py,sha256=I-XSKW6W0KpGNH9EIJjeyBj9O35Jbsn7d7VGN56zR_o,3674
optax/tree/__init__.py,sha256=WhO6d-vQIcZRAa1vXN1JNPIemEWl2I3X20FhiwWy61U,2340
optax/tree_utils/__init__.py,sha256=Ow2BN6QuNiLs5P0q4t-rlAsbjFf9WBQ7qWDRjEXwwTA,4474
optax/tree_utils/_casting.py,sha256=21-BW84YWNmJbB1dLbsgiyS-Pw1or9w1G0QB5dZKGS0,8091
optax/tree_utils/_casting_test.py,sha256=B82wFTldIBL9tHKQ410OAwzx_2UFz2Ykj0TEak2pdlA,4619
optax/tree_utils/_random.py,sha256=aTvnLKzMi_4vuXqhR_fqKjIxhJ79ncYXiq94RNXsmGo,3724
optax/tree_utils/_random_test.py,sha256=LPYcJCuzekM8OxZbK9ZKBASRj_G1MONFw_fplEy3msc,4003
optax/tree_utils/_state_utils.py,sha256=B6s-clKd2AO61As3G7a1xIgcXs5XKUhtSL0BRISDQVU,27456
optax/tree_utils/_state_utils_test.py,sha256=4GDOb6w1cDTmlTOB72dJ7For1XhtD03YJYBo9VT74kE,20246
optax/tree_utils/_tree_math.py,sha256=PAk1Quvii6D3YM-9S7tSgg03WnG3Z2Dg251iAuxXwOs,13466
optax/tree_utils/_tree_math_test.py,sha256=H8AO1OZQ7pMOTX5YzfTtk2UdDtv-7ohQuPs-2ns2RZ4,12657
optax-0.2.7.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
optax-0.2.7.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
optax-0.2.7.dist-info/METADATA,sha256=-FCUMKiHvg1QCr4Bpn-0hGe4B69RSTnbxGA79GKjLDQ,7882
optax-0.2.7.dist-info/RECORD,,
