optax/CHANGELOG.md,sha256=rQJOkWd90yYpa48lj4ka9LzlBnPaefkbAzwKUWlQ1Bs,5507
optax/__init__.py,sha256=fgqb9lcSaflZ0U9Qn4ZnQ6L07MHcrs1jVjbiBvd8PZ4,18252
optax/optax_test.py,sha256=T7MA3uQr8f74509AAtvXAdcxpj9-GeSpr8yh8N0N6Vw,1079
optax/_src/alias.py,sha256=PxVIp2QcJFEn8xbH_fesNANKT0Pmq7E0UP0EQ0GJuGs,113076
optax/_src/alias_test.py,sha256=JAItExGNtcEuY4odtybQ3aaAd9VXvg19tUqnRuzWthg,16696
optax/_src/base.py,sha256=1UB78eUSK-tZupqe7OldrcvEvWSZCXTnMw2e0xvTTcY,12536
optax/_src/base_test.py,sha256=9NOo3M0PeqUXdPPl3vugRWrTBr3Qy57_oA3AibSaA4I,6246
optax/_src/combine.py,sha256=wyD9MPOZMlB97A73Ns_JsEk9btsEfcSnsL9Cg1ZZps4,1000
optax/_src/combine_test.py,sha256=Y4wXPRnCuF8iS3n67-TnZNeyXFxKK0WBil064fAG3oQ,9292
optax/_src/constrain.py,sha256=tYrEjizwVIqtvkXZS9kLf_ZAtmaeqGhXH26dxfWzEDw,1016
optax/_src/deprecations.py,sha256=qJFeHha-32nAjV6IO7pPb1TBeHDi8lA2GxU3TxaXv6U,3116
optax/_src/factorized.py,sha256=y3NshFPWWTgvHGWuvkSPTFN-fRAaq5QLS6viNCyvK-8,8452
optax/_src/factorized_test.py,sha256=2BwYmey8cNtnwbT25IDsOpodTHOs5q1gO-PCDb2sqM4,3757
optax/_src/float64_test.py,sha256=iJ5P003Pj-EZ40oYSyGabjAeNhG-CyZF6b-46Bf3SyQ,3780
optax/_src/lbfgs_test.py,sha256=GSy-wgX7OjE6HStaONR8ZQVpGhb5yIQ5sPDQz8Pyu_A,23182
optax/_src/linear_algebra.py,sha256=6l_QoWUKu0nE4IsMktJvBiwZXlBic8rH3Y5ZTQanGg8,12352
optax/_src/linear_algebra_test.py,sha256=CQUHyhAdE_3cDRJN4oXp9NzCm73Sbe09uGvzW8nZwmU,8076
optax/_src/linesearch.py,sha256=kS6Jl8riW7BRUDvGJ6NOxbe4Xc7CsmrYNynr_Hatm8E,71363
optax/_src/linesearch_test.py,sha256=MMJ3fhFYwWrhg1fgjSWXE7I7bD_ItUIgUdCQljL4Fg4,28812
optax/_src/lookahead.py,sha256=4xwk9l_HVyXkeZu4qttsN9kdtfA_Wwn9-BsoyQOlgnI,8481
optax/_src/lookahead_test.py,sha256=v7REaBww9MDX2CTSdzGtjrHKmjHWj6l283dyHl9Nkbg,5556
optax/_src/numerics.py,sha256=_G_VNciaEOVz8uir53wigsHhA3Ps9AjfCLuP9jjRT9o,7243
optax/_src/numerics_test.py,sha256=vQnXFVjb6zxbE5u3Fql9f4ZFz2KW3vm_93eCTgHh6qo,8135
optax/_src/schedule.py,sha256=-JrpIPu2wwLPDsi-oHejHSsCT_Qz9K9Idd9IfwLmVLk,1980
optax/_src/sharding_test.py,sha256=9P1nZRE2jrn1iqSXHz-vAjwW8P4rK6HmKjuWUWaRIW8,5348
optax/_src/test_utils.py,sha256=5XfWJR9DQRFTSPel2jOLOeQB4MsLayvuEAsnJhAZNYk,4021
optax/_src/transform.py,sha256=xySwoXuM3xfJ_kRHVQIrUtvgpL6tWCrY8OWNuAW2Tr0,65355
optax/_src/transform_test.py,sha256=s8_FRDVBlnMHKRfKKjajy5ovZM8Fu5Ft0Pje1FGiJZ8,11891
optax/_src/update.py,sha256=BzEIuHWtROJG1HKZrvnhENo4oLeT4T5cfWaGyFTGfdM,3891
optax/_src/update_test.py,sha256=L43tTOHOX8F_pJ_nDbAAYfC70p2SvMETtQzNYmThCLo,3185
optax/_src/utils.py,sha256=iGNkVp_LkyCyMt8G4NiV03waDYeUVHpC6-Vvc1A8wHw,7593
optax/_src/utils_test.py,sha256=biqSoNonuHRdfbAtqqESjH1vgviJqcuD3TudUD4K99U,6796
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=NoT1ts2Ju5tL8m66xkYp8kIMfyeT0Z615FEWZuvZ7Oc,7174
optax/contrib/_ademamix.py,sha256=w6s1H33XZbpecl1a3ArNcgJdUJsgIaR3OzmEjvEakZs,19226
optax/contrib/_adopt.py,sha256=rcFTCYm9EJUpUN6_OoXO2ExwOGQbTBWa50oBMdNKk6E,6841
optax/contrib/_cocob.py,sha256=0ewB74VD-DZElB_Cxx6woa6HDBck-LFSU52yXhYU4kI,4657
optax/contrib/_common_test.py,sha256=OxxytbO-VK0GdyVu7unSXr5a__X_9HIXCC4_6sasJIY,23230
optax/contrib/_complex_valued.py,sha256=er8XuZYyBI1oDzArQejWRtFY-5XC_WJ5BdqRP_6JaRI,4119
optax/contrib/_complex_valued_test.py,sha256=cVUp7r4qm8DRrf9n8f2HAV5j3PZzroam0uP3gnr47MY,2885
optax/contrib/_dadapt_adamw.py,sha256=nqlzWheL0XmAJj90UX2VefOj9G4IsG5WsHLj-vUMUHw,6771
optax/contrib/_dog.py,sha256=N9Td8gLwoAi1bJeKp10UWFLUs7C3YWi-RbetsC_XCFA,12769
optax/contrib/_galore.py,sha256=DroyTW1JZVM0fV2plKoYj6VB009Iu1ObQsSmzKK7zew,20854
optax/contrib/_galore_test.py,sha256=z20vOvmIEVdlduM5R27eSg0jxKWdJfkKrI5XpQfQgfw,15241
optax/contrib/_madgrad.py,sha256=evM0KCSNsnhqZwC5HbpsPol4LVRx4D5vgc0olRk2Sjs,5514
optax/contrib/_mechanic.py,sha256=S3ze0uOOVESGZvVHmNXVBvK4qPeCghoMMD7leoAbuMo,8236
optax/contrib/_mechanic_test.py,sha256=mKe-ipZZzy0SOoMO9dUfS9F0CQoQ1uGlm-wqT_2khR0,3721
optax/contrib/_momo.py,sha256=_sXSN6aU6jhTQVLci-djiljAd6S_SjIVG4FJXGYMN30,13489
optax/contrib/_muon.py,sha256=eZf7UQPbSVzMyVkxfhJo3MeElMRi1ucZMl3Q0zpk3u4,29105
optax/contrib/_muon_test.py,sha256=xiKqS7iw2r8CpCtBVXjjojqxChTv1pWW57s5Dp5RRXQ,18050
optax/contrib/_privacy.py,sha256=s5W9fJn7RGvpMtWr8uxbpopI2DikDBpYbuJlY2kwXtY,6306
optax/contrib/_privacy_test.py,sha256=zNxtVqUPLcCzU7g-MjKY6ngwvrZRyl2Np3KPL7ZAkCk,5304
optax/contrib/_prodigy.py,sha256=i7E9mQXjiVk_RBl2M3rBz0R0i0PKm434_YxoTDc4Nms,7733
optax/contrib/_reduce_on_plateau.py,sha256=IasKeJmXezxgMcsdKbZyBjqiZTeknd8nxEXrQl90cvM,7215
optax/contrib/_reduce_on_plateau_test.py,sha256=y6lwHM89C_3gbswErwr6B5vTuwZ_IZTXb3M64iXIWVU,6565
optax/contrib/_sam.py,sha256=qM9MouYKIx7Gh2Mopf1kwNfIrwsos8FIO7hSrf78mRM,10099
optax/contrib/_sam_test.py,sha256=BOgVFaX2jJUU9MumpjuftivJAx7QAiEIVJmEyKBcWHw,3275
optax/contrib/_schedule_free.py,sha256=Jzvxo04a6dzUKWyu54BgwPnBjlTsD5-DSWjDKH3nnkU,14831
optax/contrib/_schedule_free_test.py,sha256=yzpX3Axs7fNq3fCB1I8LRsDKXJ1eMIPzJGxdfWlE2jE,5417
optax/contrib/_sharding_test.py,sha256=fPSKfQBVGKiW-Jqd95HL7FfMgA7x4e6_2C4_kbe_MDo,2667
optax/contrib/_sophia.py,sha256=70QsXZZd0eTshNi8MsrWz5ofvPXTMTOQibUrW8aCyrc,12613
optax/experimental/__init__.py,sha256=Fo22nhXr7Y7F7snkiu09J5Q5qbZzD_0VBYLaMVH9c-8,809
optax/experimental/_aggregating.py,sha256=pnWA4xMsUVYNEHQD890uvrWHTuCbsZtDbJZqO6p17W0,22801
optax/experimental/_aggregating_test.py,sha256=TFMIR39pl8sJwFSqwx7a1YWF4n5GMFJ5xH4jj9ri5IU,8434
optax/losses/__init__.py,sha256=AN1kSGzYouPY5RwRjMl2SMyENgz9uDAwH2jE41tVxsk,2743
optax/losses/_classification.py,sha256=TYkkGaudNdaxVGzm-HJXvOH_GOmtrlcQrMNuQzvteAI,36854
optax/losses/_classification_test.py,sha256=biKVANmT2TvTllyI3GTi-gn6WoV9Vkj9VvT9WbAy6lk,39698
optax/losses/_fenchel_young.py,sha256=eNhkJh1vwITDiAYkj99Kdr-j4QhWDvRAPuM9pjtDYnM,2113
optax/losses/_fenchel_young_test.py,sha256=Dr1GWzsoPzDOPoQlaBhWRdUy8duPKbTlxKc6uDZfFAA,2688
optax/losses/_ranking.py,sha256=GiRhX5IW5qoWGG-q5apEEE_l05-zL6R3FnflFrZAwIM,6408
optax/losses/_ranking_test.py,sha256=l5j98ik5-sVfbBa6GSlffmtnVb1dQ_XbLU1mxmWxs20,8941
optax/losses/_regression.py,sha256=hngmRGW0crZn9vL5cVclteEw3lqrru_jHh_tLNBJQSY,8756
optax/losses/_regression_test.py,sha256=rp9EqVAAzmUjgQv2jaF08mn_FFU8DSY4C4FZAlnNDcU,11148
optax/losses/_segmentation.py,sha256=6fbv90VcwzeTZNo1TZnVfCdCNaspPGx_sb7tKy-CBfM,10680
optax/losses/_segmentation_test.py,sha256=uh9sm-811mjiKAyHw8jtuIkG5d9Eq-tF9fPD3jDKnJg,20235
optax/losses/_self_supervised.py,sha256=4-c_2yhjh6cZHNq85_wc7cH0et_5jFBZPYlf8tbr5F4,6966
optax/losses/_self_supervised_test.py,sha256=vUN3qXYAFY0A4_jF2Xf9YwGtEY5YvS9Yb6G31K_pEM0,6400
optax/losses/_smoothing.py,sha256=yrXVg4KvwhsQjT5GrrbgTGTzCY-ThualUeEwTU3HbmQ,1915
optax/losses/_smoothing_test.py,sha256=WE90aVySJ0Wp5uFZmqjzCP2IHWHFAxBUGuMOtz-KFYw,2333
optax/microbatching/__init__.py,sha256=gsGyT0AVdRv9_bzR1sBNSPeZPftXoPVgyqD5pJIAEsk,1052
optax/microbatching/_microbatching.py,sha256=qAjbAHdACGr-qwpbms89xDDPvPmMVjRzbOUjiTWHrGs,29414
optax/microbatching/_microbatching_test.py,sha256=J0w1YvL-NeMTFlXeQRvoxV4p4lMIy1o4vz6iK2YEDeI,21798
optax/perturbations/__init__.py,sha256=VMj_AXmy85zJS4FmyljYqwtv5pQdzBtT9eQ1IQmUrsM,934
optax/perturbations/_make_pert.py,sha256=HulzoGnblqgcaQz2W3J03DqklYnTT6VpGDLCeyRGMwA,6336
optax/perturbations/_make_pert_test.py,sha256=uSRnk4aKCR8l9iRt5MvRMI2SNpcnJBZZ72yLInuvzfw,10165
optax/projections/__init__.py,sha256=NOIj2H3EW78oS-x3TrFbp3HsdulkOQBgwlF5vgANeww,1528
optax/projections/_projections.py,sha256=1nYKp9GP9Fih7v5X3EYGx41DHtpCpoc5bOZY9MqwcWg,10139
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=-SrEbBeF5_hTbWeNzBuVfDi4m4N__Bxvf1V-DxH7K04,11128
optax/schedules/_inject_test.py,sha256=Q1WoCWLuCFfN20MzIRboh-7hqHsGJMyrxQ0Bw9K-rUA,11884
optax/schedules/_join.py,sha256=NQmX1LWzMDtQH7jDCQG8pcB9xjlLDVxSqywGhxU4zLM,1706
optax/schedules/_join_test.py,sha256=NFjwadA4Udwd1bDmeJMBwa0tnG8bmuDiQsidehD6zLQ,1654
optax/schedules/_schedule.py,sha256=R_9ZXZNQGXWEe-uOMt9LcNynhi2MLx5JSVgPnFqm_tA,26373
optax/schedules/_schedule_test.py,sha256=7YDyggPUiRWYot4VaqKsCP79mdIJIajWdKcxsykIJZ8,25116
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=asWeMgSD3n95SA1d4GBZEEBojtIjmi7_5UkgndPYzf0,16079
optax/transforms/_accumulation_test.py,sha256=Ozas6wEsCQdntKjxpDZ5NgbS9K5vkjCSSVILXHsD3B8,15777
optax/transforms/_adding.py,sha256=FXAo_huVsQ7JzxVyvzSjy1sqzbmNUnR263UhPqX-KTw,6010
optax/transforms/_adding_test.py,sha256=JTJZuOZDZX6YgWUTJbP0hws0VR4ti8qkRIG0116tYpg,3721
optax/transforms/_clipping.py,sha256=fAOXd6EAszrirEfroTrsYbezqw3Ntf1UykoJFmApFJs,13345
optax/transforms/_clipping_test.py,sha256=xI9crIixEPZE6Z8scpLPKcRWejfNPtYmHaizVTecJi0,14493
optax/transforms/_combining.py,sha256=WCKet-crqgVkD-EfwtF87gaA6-dVs3Y6ppIv4Kfgd5E,11182
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=iAvJt7BMoA9QTlaWr3sRxC4jmDz3yWkTVoc2cRjMcjM,3334
optax/transforms/_constraining_test.py,sha256=-zxlsS-y3am9dl9umZOXko_v_K_7NHHujctLyjCQROc,4671
optax/transforms/_freezing.py,sha256=gkuROQXgVuaXmFfIaoYNV4a04QqYsghw3-IG7RNL-YU,3847
optax/transforms/_freezing_test.py,sha256=BCa6qB8ftuyTMOXOVqJf2mPwHNjrz-8eIuS0rq8mhJc,4389
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=Bes76ztaSfjMw4k3feYC3h7mNgHC4uYVj0MesP0vMBE,6305
optax/transforms/_masking_test.py,sha256=zU0hmXDiAAMgQUKwsD1vqQtLaha4k-fwNXJp0U9efT0,13183
optax/transforms/_monitoring.py,sha256=hV5SrT09yOhZIUY6ZwZmuHKuHSrCvVOERnCC67rIFsc,7669
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=o9sbZ_aHnMwA-epyZqbFJgY4iIcbyroMa7uYzhR4sSc,4619
optax/tree_utils/_random.py,sha256=Ipb-DnlVg7Awo3HsQylFVto0JS7FWknNgI9AcHRcdy0,3736
optax/tree_utils/_random_test.py,sha256=LPYcJCuzekM8OxZbK9ZKBASRj_G1MONFw_fplEy3msc,4003
optax/tree_utils/_state_utils.py,sha256=wdtDrsXTKGvlqKY_bkAq2I7bRYz6vKYwzGemyAP8sQU,27522
optax/tree_utils/_state_utils_test.py,sha256=J14XefJ_QinfePt5j1yCnuDp24HggEW2lyzAv1O8Kxo,20147
optax/tree_utils/_tree_math.py,sha256=oledFBAH0txM3wzK_38A1vcN0gZwYp3Nlc-OufPPakA,13446
optax/tree_utils/_tree_math_test.py,sha256=0xGt9KqJtsI-l1G3fDHi0O1PbwhX5SsqXlqDATWyGJ8,12690
optax-0.2.9.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
optax-0.2.9.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
optax-0.2.9.dist-info/METADATA,sha256=sEk_E27IFQBR7fK5OJfbrOXtgbEUpKahk4aO1ZjJH2A,7882
optax-0.2.9.dist-info/RECORD,,
