optax/__init__.py,sha256=XI_o7EM9lZwdvUqFI921dfkiCux1mOeH40zwzSz1mVk,17611
optax/optax_test.py,sha256=T7MA3uQr8f74509AAtvXAdcxpj9-GeSpr8yh8N0N6Vw,1079
optax/_src/alias.py,sha256=EWQXR7qdEiMRZol3jRDnUyGFSf_xhzBVIUnKXD0IhQ8,108661
optax/_src/alias_test.py,sha256=EYKlXCv0KfkmiqoMf3r4Z2wUKr-In0fdL6oJDwBAH2Q,37400
optax/_src/base.py,sha256=2YRyc0XAk58uUozKEyRuddNCzuh81_anjc_BVQLp8m4,12312
optax/_src/base_test.py,sha256=s10sCATegmQHrrpCJGPjcOp4OKPpCu5giKfNlqKxHeM,6561
optax/_src/clipping.py,sha256=2b98Edu1waORroKAjZmbTIlR7cAxJMZJZBE6IbKmJrE,1305
optax/_src/combine.py,sha256=wyD9MPOZMlB97A73Ns_JsEk9btsEfcSnsL9Cg1ZZps4,1000
optax/_src/combine_test.py,sha256=BGvxP1kguCVgx8hUZ6mV0JFzziEYNed5tg2oAfzMAD8,9315
optax/_src/constrain.py,sha256=tYrEjizwVIqtvkXZS9kLf_ZAtmaeqGhXH26dxfWzEDw,1016
optax/_src/deprecations.py,sha256=xOzxskYhj60JwJBvgKoy6IGpLlj6lrKGm52wPU88Mww,1938
optax/_src/factorized.py,sha256=dSrP1I2skCI0uO80jCcFld_ee7gAAI8C9dOabk_cBss,7734
optax/_src/factorized_test.py,sha256=yxKAofjiUa97XSXjWOBS20PUz3DSD9o5wf8oTqiYbuw,3614
optax/_src/float64_test.py,sha256=YNJ7G58QXOcfZQif7kOJ3WjS6txD8jcREy9Su2RwtF8,3807
optax/_src/linear_algebra.py,sha256=ZGgRuUHjHhuUa9tGF4FzPsYKSaCJD4Q1fzGPjkSUY_w,11846
optax/_src/linear_algebra_test.py,sha256=dfYuGSxuA5981beskwXOfUvm5Xf_lldXRby1KHYAUmQ,8277
optax/_src/linesearch.py,sha256=Q38p3UH0FNrnw0cn_9lZtYkv75-H5MnyyDmuEknfZPY,62822
optax/_src/linesearch_test.py,sha256=R_mLbF2bPqwZmE12XTgCoeD976jckvqvqvUIbZb82QI,25209
optax/_src/lookahead.py,sha256=KL6XQxN2QWxdCEbjBnOa2qJeSi5DnVoROKp3YVSjzr0,7581
optax/_src/lookahead_test.py,sha256=0G2mydb-wpuFYPWZvQ8QEgD-oIzhut_xd4ScrvWxj5g,5654
optax/_src/numerics.py,sha256=WKDbu8kWqds7wlZ0Rp4gt4wZUZGw2jAeGsi_0Jl8Rjg,5424
optax/_src/numerics_test.py,sha256=2c_NZBv0irBkmIpmaLm8GmyrAVL0pvwc_5a6l0niPRs,5605
optax/_src/schedule.py,sha256=-JrpIPu2wwLPDsi-oHejHSsCT_Qz9K9Idd9IfwLmVLk,1980
optax/_src/transform.py,sha256=pATI90gPTEsLyPQs3BBsd4OZb_nYyZgoti_15e0XET0,60616
optax/_src/transform_test.py,sha256=-4-padKSZV5Z4vRsYTZaWkQSEbbirDIEGm_CQgyM-0c,8004
optax/_src/update.py,sha256=n3qE_n3t4D7713RERwe2pil6UxPlzMYzsXqSk7-B3TM,3851
optax/_src/update_test.py,sha256=2Mnco4ae13nIapgBUvRHf-UxTX7i2XDFTrfJdsvKA7U,3192
optax/_src/utils.py,sha256=ii1TkrJT7hvgmxfTR753H5fvCh8HgU0IAxD2wCt9iu4,11587
optax/_src/utils_test.py,sha256=VhQloVz1XJKj6UjR7CeACw8QKlYSD-POcS7dAKGs-p8,9247
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=ceJbQAgI5DTHyVI-7LcPL8mFT72KwmJb9Kn4MMWMWWk,3336
optax/contrib/_acprop.py,sha256=MILzRF26PsML8tZulgdz8XzcoMrNUYhvtVZ8F_4U8wY,6872
optax/contrib/_ademamix.py,sha256=62ki4SXN6oagKL75gQv0DuZ0pUr4YGGhwrYKPxnSbT4,18532
optax/contrib/_adopt.py,sha256=Ki_SnxbzL063YoF3977TFUtelnF3mc-L7oK7TlO4Exo,6711
optax/contrib/_cocob.py,sha256=CgslV8S-cjfk7x1YPE043Nj9NwAZ54X0rnP-qlnmJGk,4494
optax/contrib/_common_test.py,sha256=HhTe1OUDRLlBm4fbRCfUkmKTV6d5GDFknZbzQWXBnPs,21353
optax/contrib/_complex_valued.py,sha256=xUuG3dB7rT10SLjBvZ2IaTBgLhw-HoA2Oapnhd8QdMA,4071
optax/contrib/_complex_valued_test.py,sha256=RANkqcoN8JUYRIJfMbRiRoTfkyadUo8_TXGGTe3MwfA,2924
optax/contrib/_dadapt_adamw.py,sha256=G1MewRQ7u4KxhozwKNocs_UkBE9j9pAiGuz0J8hzJP4,5565
optax/contrib/_dog.py,sha256=9qTs99DePNAx_BSm27yilnET7deKaPzqfnn-NOnTxSM,12450
optax/contrib/_mechanic.py,sha256=i2eEQNB3EMDSgHOODrrBrejlh9IwrVOEaoKtR7uR0zs,8024
optax/contrib/_mechanic_test.py,sha256=cfxuFQ3i8NckIvEu-4E7XzMGtB5lGUtrFoCN-po8kks,3729
optax/contrib/_momo.py,sha256=7JoMXXSIFyt2JfIYeld_iVykX5-6GmiWItX3aGFlCg4,12933
optax/contrib/_muon.py,sha256=fgTv873rDh9wmiTllGJWSC12EQstMg_oEsLjNZbCoIE,18689
optax/contrib/_muon_test.py,sha256=VYvDwMPgiSXT4dlIJe7c4efeBpdq7znYktxGLLqgOks,10061
optax/contrib/_privacy.py,sha256=jPTE3NsUPAw6TgJ87l6mMNG7Eg_3IPc9IYwA39Ldqgk,5695
optax/contrib/_privacy_test.py,sha256=EwT4XKdUP5Bc0Qhcb6pQDzgdNcrWX_5gH6Y_pxOL7qI,4401
optax/contrib/_prodigy.py,sha256=eDXLCeLoYRSKgnkRIrEBwO8mk7dGBzw081yrKqbVnD8,6447
optax/contrib/_reduce_on_plateau.py,sha256=o81-ja9jfd3Ktap-glei4F-T3VVWJSgVb60YqZ1Mm_I,6657
optax/contrib/_reduce_on_plateau_test.py,sha256=ukw_HFxeJEz-1Ihdel2GBG6YDN8vmuk4Zn5af2vEAtU,6417
optax/contrib/_sam.py,sha256=xt__nccpI2L3nlEJy1I3qUuQ-aD5_GaXifPYNw29m_U,9960
optax/contrib/_sam_test.py,sha256=XWQ4nWADcPwi9TyLVMovDMCKt6ip_9x6YvdQ38E4HE4,3229
optax/contrib/_schedule_free.py,sha256=oCWZojYIwaEXBQ-c25g5Hj15hyp0ymvZnXN2sYhfYEM,14158
optax/contrib/_schedule_free_test.py,sha256=72aRH8JkgRgqSXovp8q1LrTrEWN0YqjRyAcGK4FPURw,5147
optax/contrib/_sophia.py,sha256=guS4U8GdQ3X4mnkR_Xb0mG-wrds2b3DgWhimz48sQFQ,12134
optax/losses/__init__.py,sha256=ivaiSJ1IE97CFXGSd4P01roFXdljgJIPLejEF1vy96c,2676
optax/losses/_classification.py,sha256=AgWh92o9C70bCyU2E3LceO8XGsRaCkMM4bWECFkMZbM,33261
optax/losses/_classification_test.py,sha256=PXwuPGIkwqKu10aWgdBlNcUU4hR-rAjIzn1yOzqcsVY,38713
optax/losses/_fenchel_young.py,sha256=mvK8m2eUAF3CRiTI-UYCvviM-hLAxUfDpkLHMPgrJY0,2106
optax/losses/_fenchel_young_test.py,sha256=omVS4hCZgLfabIgEj5NfLtfNt_BvnoTyROFiUSpKA6M,2661
optax/losses/_ranking.py,sha256=TDsExkcN16r4lzHPO0mNd-WQAhIWrgvymCfcy3mJnHI,6231
optax/losses/_ranking_test.py,sha256=l5j98ik5-sVfbBa6GSlffmtnVb1dQ_XbLU1mxmWxs20,8941
optax/losses/_regression.py,sha256=smHbXs7eWPI2zhxg0xFugxeE2JTBTZf3Q82QubVMJFA,7803
optax/losses/_regression_test.py,sha256=4E3NcHeP2dZieLCFNepRF8jU9salrSYweF5Pbk8nTNU,7201
optax/losses/_segmentation.py,sha256=-mERuMBIrhTsjCRhZcr6a1nIGcZBx4oQsm8AN9f8MuQ,8916
optax/losses/_segmentation_test.py,sha256=yDUSLmcyN6xX-D4BsVXJMC2FJ1kdsyEJjwVlLyb5-GM,18474
optax/losses/_self_supervised.py,sha256=0mPk3ghNBpUCgilzh_wwqzb_MJjGy5XDimByLse_RVc,6201
optax/losses/_self_supervised_test.py,sha256=JpK75SkYp1gm2nlT2H0a0W3AqlqvyhReHG4aqLH6v9E,4232
optax/losses/_smoothing.py,sha256=1Or63a5tROnvMJ971j_t1kRpU4EZqplWTJFVBKhodt0,1495
optax/losses/_smoothing_test.py,sha256=jGG-Wgaq_lc7hAaV52lCQFMPkpulgLXPv2Kem7odjSg,2450
optax/monte_carlo/__init__.py,sha256=83JESSElEeZYnGwiCJsoiWMMz-9xrX_LNIZ2ABorbfI,1349
optax/monte_carlo/control_variates.py,sha256=19t2qhay7mHO99H1qVS1tgcqD_Gv0ePpovlYILjW9VM,17916
optax/monte_carlo/stochastic_gradient_estimators.py,sha256=4galLjfe_4NXmTZ5fmyoM1LBMSw4qp0aYgWk_aetzIA,12851
optax/perturbations/__init__.py,sha256=VMj_AXmy85zJS4FmyljYqwtv5pQdzBtT9eQ1IQmUrsM,934
optax/perturbations/_make_pert.py,sha256=BP87AKOsg9Ka_0TsjJcREzasfO8M2w2BmOQIccWunwA,6298
optax/perturbations/_make_pert_test.py,sha256=BJug7SZVj_VZFhMqekd5rrlRYcEulyJozaoRGSIrGYU,10147
optax/projections/__init__.py,sha256=NOIj2H3EW78oS-x3TrFbp3HsdulkOQBgwlF5vgANeww,1528
optax/projections/_projections.py,sha256=NduTDq7wORYn1jyOryiqmYthvUlSLJqtvPJ0QggELGo,9878
optax/projections/_projections_test.py,sha256=sFcXNAXJWjTGdyXGmtGEcNFP_ruldW4ozXHDbO4lbP0,10047
optax/schedules/__init__.py,sha256=Co2sonQPd-qNHaeJieqCRZAqvKYXzGrmoDlGKh4-p0c,2002
optax/schedules/_inject.py,sha256=k7k33ihmZOrEs8Q8ilTItIoWkZL2rF7jPSMIPXB_Nks,10428
optax/schedules/_inject_test.py,sha256=X2OrHhmJTK09R_kr7BP0yRjMyrhzsRbdeJxBVNq7Kv4,10920
optax/schedules/_join.py,sha256=k5UengpNrjiXPXa1SMSSp2uSUQYf80crjouWhjWFiU0,1644
optax/schedules/_join_test.py,sha256=NFjwadA4Udwd1bDmeJMBwa0tnG8bmuDiQsidehD6zLQ,1654
optax/schedules/_schedule.py,sha256=aVOe6Iok_GFayrj20zqXk8-4FnYrsqBtI5l2_JBKyVY,25668
optax/schedules/_schedule_test.py,sha256=KIf7lcGFlkyABWEwbFWvIO7kmbYH-WglU-DPnEp1pns,24084
optax/schedules/inject.py,sha256=r-Zzd_oyV19qOr4m4jfePuI5y8Ih72_tpjo1zjyQ6Xc,897
optax/second_order/__init__.py,sha256=R6ir3MXRErlQF9GT0SKbB1xlsKDCKVA6uzYojmp7Hi0,932
optax/second_order/_base.py,sha256=b4VB6QTqMwKzVgfoVFaAqbD6cew08SQGaUKHjlTeofc,999
optax/second_order/_fisher.py,sha256=DT7sLFWeybN9ImW4wofBJ9Grue_a96GQBsPFfrARY3w,1948
optax/second_order/_hessian.py,sha256=vqpwS5t9NQfeAMG3L890Y6J4VPODEEXteTCUYotgkv0,2528
optax/second_order/_hessian_test.py,sha256=qc1ZXpNhbdLQBgF0rxy49Q6-D7rfIHwLS1qrsCB7_qg,2701
optax/transforms/__init__.py,sha256=d2tD9WV5xntiMBnqpe1Cz-u7YT_cqboP_gD0f-Xp8JM,4006
optax/transforms/_accumulation.py,sha256=sOL6SQIMOjGHwWFHjKMtsg9_rCulQ-KWrL1UHuvZlq0,15585
optax/transforms/_accumulation_test.py,sha256=Ttg8PFh5uarSBdJiS_I7HqQ4UCDba-kBm1LvAugAoEo,14207
optax/transforms/_adding.py,sha256=tndvYJlNefQ8LXMQLqvcIoVjij8TR9pRj0Ke1VteU0o,5601
optax/transforms/_adding_test.py,sha256=1Wsr7T-oGDS10Aw3syEYZVLsy2-x3XtvkVCptoCYTUA,3720
optax/transforms/_clipping.py,sha256=PT1mEy_VqOlcX2pfFm7WsTBMsbv025z4hnqYrfOGKvQ,12527
optax/transforms/_clipping_test.py,sha256=_TE7Uf2gW_yuJVlwaO7gGVqQz66O8im8G667504xsjQ,11359
optax/transforms/_combining.py,sha256=O5kuGpb8P8N9jlhfG2xIO--wSJZypanMdlOX53liFSo,11134
optax/transforms/_combining_test.py,sha256=zDl7XXyl5bpZEN1kTi8R1Dfvc25mHtIE2AQrqqBnuJE,9355
optax/transforms/_conditionality.py,sha256=zuYcxVa2BeqhWxnX58mptCJ-STwP5j4qn8c9iBYkslI,9421
optax/transforms/_conditionality_test.py,sha256=MHSLYnZeuxYaSUbOIzFLtRnkSNAXMrUq3KDxFl7TLJI,10648
optax/transforms/_constraining.py,sha256=r7zzlSOJsu5Y907IxXN6K540m2FK9_p2ZvPIkeW51Io,3256
optax/transforms/_constraining_test.py,sha256=6wxOHX1hcH2W6SO-p0LtZc0Lpbi2gxsumWCUCOCz1_0,4609
optax/transforms/_freezing.py,sha256=qYNcIuE3GZ_HJHM1P_q3cKDjp9OmdOIllSJsEFz_zvg,3858
optax/transforms/_freezing_test.py,sha256=_pxtRcC0iFtISx4u8j5SYSriXWoYKEM3tGK0ucETghQ,11824
optax/transforms/_layouts.py,sha256=fk-z2A_BW1rGt8GsnwqQ3t-4GKsmnOqcKenNvLhsfCg,2616
optax/transforms/_layouts_test.py,sha256=KcGLp0D1Eu7iQ9d9O8_h_tooa08x2cgHd9f0mdYXg38,2020
optax/transforms/_masking.py,sha256=mqYJHB99tJtYtn8MczNejKevdfInH6wjE6rxzFr3XSE,6262
optax/transforms/_masking_test.py,sha256=Sh_nn-dmJSqSXI1odbZbGbzznh4yawI458DpwEEwLi0,13307
optax/transforms/_monitoring.py,sha256=OQcq4msoJH7nqLe-GUo-xbPQRjaIdl3KdX_owmzK4lk,2912
optax/transforms/_monitoring_test.py,sha256=LxeVpD0eu0pTDNMq_DrbJ1pfgJfkGJ8c5F7XxwoHzXM,2165
optax/tree/__init__.py,sha256=WhO6d-vQIcZRAa1vXN1JNPIemEWl2I3X20FhiwWy61U,2340
optax/tree_utils/__init__.py,sha256=Ow2BN6QuNiLs5P0q4t-rlAsbjFf9WBQ7qWDRjEXwwTA,4474
optax/tree_utils/_casting.py,sha256=JSGHPIKgR4sAWEsoPZHq6t4kYxuY28IuA4ZM5yfuPm4,8035
optax/tree_utils/_casting_test.py,sha256=B82wFTldIBL9tHKQ410OAwzx_2UFz2Ykj0TEak2pdlA,4619
optax/tree_utils/_random.py,sha256=STG8eLdh8_kwW2ORAAkUJJvxmObkqU9dCcUuAMpgaCw,3639
optax/tree_utils/_random_test.py,sha256=B88g1i6be-44ZHw7306Y87W4JLrZhYE3ghBvF6MhSqo,3928
optax/tree_utils/_state_utils.py,sha256=_pzPMAhbbfClCJ24N1nxQscqrVDGBUDvLwjNwFjiMiI,27462
optax/tree_utils/_state_utils_test.py,sha256=9VRzqZtJmAu7jov6gmBhdCNc1JfIUDs606z1pSH3pi0,20863
optax/tree_utils/_tree_math.py,sha256=28BuNgc1MR4bT7L9E23mCdoEcExsZxBJNzSDhMsFLDg,11945
optax/tree_utils/_tree_math_test.py,sha256=KiVSmtXaIhIXOEdTGPJI2Tgp0OX3kLhNEbUTESKZulE,11646
optax-0.2.6.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
optax-0.2.6.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
optax-0.2.6.dist-info/METADATA,sha256=jGy_0ae3-oOjxl7M4TlLJ3UHa0DH-Vxr5Ye3iSW9qDQ,7563
optax-0.2.6.dist-info/RECORD,,
