optax/__init__.py,sha256=p5iuUQhTpJQgFrGw-iii5aafgclccF7g_fTvh7q_a34,17288
optax/optax_test.py,sha256=T7MA3uQr8f74509AAtvXAdcxpj9-GeSpr8yh8N0N6Vw,1079
optax/_src/alias.py,sha256=zShYqaLMq1og6cOvthAw92BIO4Cj_z7m6zNhx69pzOs,100566
optax/_src/alias_test.py,sha256=t5aTpoSunvqwO8noTbC9nukJE0GEe59v5UPp7ABxwn8,28764
optax/_src/base.py,sha256=r2FFKfUvK4HAMMbaFc-gNcir78FDcuErED4ktb2ErWg,12224
optax/_src/base_test.py,sha256=s10sCATegmQHrrpCJGPjcOp4OKPpCu5giKfNlqKxHeM,6561
optax/_src/clipping.py,sha256=2b98Edu1waORroKAjZmbTIlR7cAxJMZJZBE6IbKmJrE,1305
optax/_src/combine.py,sha256=pi3Mp19T2W7zS0K04Kgwfbkb82VRemLPyRmuJNWq0FI,936
optax/_src/combine_test.py,sha256=XQSe6X3LvCCKyhmQnTxoOL6mKuoOY1TABDqHUeIWwCU,9356
optax/_src/constrain.py,sha256=tYrEjizwVIqtvkXZS9kLf_ZAtmaeqGhXH26dxfWzEDw,1016
optax/_src/deprecations.py,sha256=5umkxmG6vo0zgdltDZsHCngV5xE6aq4l7ppl3MSZfHE,1926
optax/_src/factorized.py,sha256=A4iXrIMJb1RPim_jfLFaXVmJFTpTGUANZtkUSm03LLY,7758
optax/_src/factorized_test.py,sha256=yxKAofjiUa97XSXjWOBS20PUz3DSD9o5wf8oTqiYbuw,3614
optax/_src/float64_test.py,sha256=fZwmcpXxdXXt3r-hLmZcP2TX2VSKy5ZMDXVp2PjPz60,3804
optax/_src/linear_algebra.py,sha256=aF_UajnCHjghMAJ7sunFtXOXdhsR7bBvs3Eg09UDyLI,9556
optax/_src/linear_algebra_test.py,sha256=DaF30nnNRYzpNODtM5aiOmOikZ465nquENDev5bQsfg,7224
optax/_src/linesearch.py,sha256=srGf-17AnzeGRYK_hpmJ_ofYKm5m-lASo0ziLIefWgE,61860
optax/_src/linesearch_test.py,sha256=6B6s8pd-RIiaYRpwXSMsKL0e-QzXyRjpXTxh8rmxQwk,23178
optax/_src/lookahead.py,sha256=KL6XQxN2QWxdCEbjBnOa2qJeSi5DnVoROKp3YVSjzr0,7581
optax/_src/lookahead_test.py,sha256=hwcO4K7dPMfTvW0m4aHqNwQPqJXw8GZo4blRRR_lJmY,5692
optax/_src/numerics.py,sha256=WKDbu8kWqds7wlZ0Rp4gt4wZUZGw2jAeGsi_0Jl8Rjg,5424
optax/_src/numerics_test.py,sha256=hd3CRGXm-_wWZ36cWkWglHU3t0EAzxXnJ30KB6yn3Fk,5437
optax/_src/schedule.py,sha256=-JrpIPu2wwLPDsi-oHejHSsCT_Qz9K9Idd9IfwLmVLk,1980
optax/_src/transform.py,sha256=pIy6HlrN2I2X_yYDvIEViaKDclpniRBzYUmMHDpQWPQ,59356
optax/_src/transform_test.py,sha256=oXfFsKta7ASLuwPu21ImILqmTHlGau1Ju4IaCFaDDzA,8004
optax/_src/update.py,sha256=n3qE_n3t4D7713RERwe2pil6UxPlzMYzsXqSk7-B3TM,3851
optax/_src/update_test.py,sha256=QqITPFQvU486etw09zjMUvuVbL9sp9H2s63bdShCO9M,3188
optax/_src/utils.py,sha256=KpF7PDQKTsX2SzKRJrnoZte_qQTj25yZUZGZxikdUIA,10651
optax/_src/utils_test.py,sha256=4DDGrMWcK29wISEtvb46YZxB0ltxPqbXO81BhVYZSeA,9247
optax/_src/wrappers.py,sha256=M-MHFGjXgZs0l7AhiIP3ah4t86ysA-PZsNyI0NL-kn0,2210
optax/assignment/__init__.py,sha256=ibogr1pLDW_vlW9vYY7h-oe2smKrlmaYQipSDUvt01c,838
optax/assignment/_hungarian_algorithm.py,sha256=WdN2fbmSPlWcgUhzYJwYZEesRKEo-bcYdsI3FYxTCcM,8944
optax/assignment/_hungarian_algorithm_test.py,sha256=DNhbcnP4nWd2QpsY0iElQlZHuIcUMRT6UmxH1STowPI,3720
optax/contrib/__init__.py,sha256=EtNLWwsJkgTuBlnc2Hha6CR6FNqWY4VaWGJ9ftLIHpU,2522
optax/contrib/_acprop.py,sha256=qW5JZ_1f8WYnet5XEgdDxtB9BFRb_QrPnp-B0-yeQJQ,6832
optax/contrib/_cocob.py,sha256=cHuXB_o9OJj63Uk6N7olk6rjYiX7seT9tIxpJX37Hcg,4506
optax/contrib/_common_test.py,sha256=Iy-g4aRg4Mkc5vjVPr4Tz9MyRSpAlUw7ehaf90nYISY,13523
optax/contrib/_complex_valued.py,sha256=xUuG3dB7rT10SLjBvZ2IaTBgLhw-HoA2Oapnhd8QdMA,4071
optax/contrib/_complex_valued_test.py,sha256=lmCNUwNyXqoDhZ5E3bAN2J8qD5dOGMhKQI-ZHIx-_kY,2873
optax/contrib/_dadapt_adamw.py,sha256=nfSGcAMYU-JSnBISr3KXsf9ZmdCKo675gSGS8tiQRuk,5398
optax/contrib/_dog.py,sha256=IobsFa3_mLA7grXJ3sia0tz2Qnqw_38WuxVANcRaTkI,12560
optax/contrib/_mechanic.py,sha256=1no1o16j7iIQIaCo5tDgmJndHKWNTdk8UFE2no8R46Y,7830
optax/contrib/_mechanic_test.py,sha256=YSaKy7xCxMcHsTcz2SC9nKrWH1kQgbn0ICNEnCprZmY,3760
optax/contrib/_momo.py,sha256=zwj4i6dgNIFJ_oEj0y8aEFrueb7sZ84iO7P7yBj1Nog,12655
optax/contrib/_privacy.py,sha256=6BbSC23qAuOeLm-Tgf7HDi3cuAw2Cf15OJQ-6JfF7tE,4776
optax/contrib/_privacy_test.py,sha256=puOFSC3xF3Xt9O4a56nfWfHwL5a8yj9tXwYhCDtus3o,4405
optax/contrib/_prodigy.py,sha256=5yLVDuV8LFZNLv4cwGzS7CNV13ZidMP0b_pcFrVnPdc,6280
optax/contrib/_reduce_on_plateau.py,sha256=xOvN-M0QeDRQGCNkmIsRtnBh7yNjWwG3aSMPLC8tROk,6672
optax/contrib/_reduce_on_plateau_test.py,sha256=ukw_HFxeJEz-1Ihdel2GBG6YDN8vmuk4Zn5af2vEAtU,6417
optax/contrib/_sam.py,sha256=-2ztFtB6ShtfD0tbGoK7g87BCff2goctNWz8bdzSMTg,9963
optax/contrib/_sam_test.py,sha256=X0iktCrmrroke7cCtSXWgMP3nNNzJ1kuF7S0x3Ur5tM,3258
optax/contrib/_schedule_free.py,sha256=ItAYNVrkqWIRUYpkAr04eYWA5IiwZPRV9BTBG1KU1UM,14177
optax/contrib/_schedule_free_test.py,sha256=72aRH8JkgRgqSXovp8q1LrTrEWN0YqjRyAcGK4FPURw,5147
optax/losses/__init__.py,sha256=pLrqMNIl7_gqJ7Ei3g9wJBS1VdwaOxi-51y13M2TQZw,2360
optax/losses/_classification.py,sha256=YuJXW2KGwybkryft8eg9B-NPJKHq1tbM2N97x9mD8BE,29721
optax/losses/_classification_test.py,sha256=1lIjP0f97kFiVgsglEISiEQy5wyjiTN8FaOV0N1B92g,33404
optax/losses/_fenchel_young.py,sha256=mvK8m2eUAF3CRiTI-UYCvviM-hLAxUfDpkLHMPgrJY0,2106
optax/losses/_fenchel_young_test.py,sha256=KEBNhEO1tN-_LKM-DIppELbLt7HmdPQqrm855Iw5uX0,2665
optax/losses/_ranking.py,sha256=TDsExkcN16r4lzHPO0mNd-WQAhIWrgvymCfcy3mJnHI,6231
optax/losses/_ranking_test.py,sha256=STnvCKzZOa-0cgihImo11nBrtbbGnGVr8BFII-iJroI,9022
optax/losses/_regression.py,sha256=smHbXs7eWPI2zhxg0xFugxeE2JTBTZf3Q82QubVMJFA,7803
optax/losses/_regression_test.py,sha256=iXjSUkqacODmpDaLQDJHgZj4-UlB25u_XvthVlpsAOc,7199
optax/losses/_self_supervised.py,sha256=Z9JpMJRSlfeB123wYVID_Vp1RDKvWEoDLVRnEUg1g4g,4004
optax/losses/_self_supervised_test.py,sha256=wV5S61yRpAm5hgpVuXhMnDpe7wgptg9DKsexhrBRv_U,2047
optax/losses/_smoothing.py,sha256=1Or63a5tROnvMJ971j_t1kRpU4EZqplWTJFVBKhodt0,1495
optax/losses/_smoothing_test.py,sha256=jGG-Wgaq_lc7hAaV52lCQFMPkpulgLXPv2Kem7odjSg,2450
optax/monte_carlo/__init__.py,sha256=jFPl_-QVH5GIuDKEK8a5FdRLuVaX78gZdFhjUYi6R0I,1258
optax/monte_carlo/control_variates.py,sha256=ljnlv1xM3xTL1dIEdiGgTPtMZyt1AVKHIwHtoa6gjDM,17882
optax/monte_carlo/control_variates_test.py,sha256=VMjcCC53x83IXs5BkRqzMsTV0ayOlBYZW2uc0HKzroU,23332
optax/monte_carlo/stochastic_gradient_estimators.py,sha256=LZ3jgGTHaU3xgH0SEqrS_IH-mV9mOQbPYy-YO0Ag9Ok,12855
optax/monte_carlo/stochastic_gradient_estimators_test.py,sha256=-iwyARLdWrlG8-owmoGiD7pgA-jUB7HyIU5H7dPJvSI,14143
optax/perturbations/__init__.py,sha256=r24jzuoIQuP8wmxPjNgmB1oE37xkYNi-bzTTeOsUg3I,935
optax/perturbations/_make_pert.py,sha256=YRK_HHLr3PYiiJ5dxWLSeaVYIZlbN7a9sdOOFRn6WB4,6588
optax/perturbations/_make_pert_test.py,sha256=S47vImnG1a4M1Pl3vowXFV9iTzG3tpMmJmrz1SxGkPE,5841
optax/projections/__init__.py,sha256=gUj7qkLPZXGKZK1aJbUQq4DzK-maBNWGR5gKhlfiEJs,1338
optax/projections/_projections.py,sha256=f2p0UNEDTRt3OluujRVPyuLUa5z6v1n8klIDnGSLdfg,8269
optax/projections/_projections_test.py,sha256=_d4ab6Cq3bOJxi2mz7Hf2vTiYTM8kVusjkALayWJ8KA,8337
optax/schedules/__init__.py,sha256=Co2sonQPd-qNHaeJieqCRZAqvKYXzGrmoDlGKh4-p0c,2002
optax/schedules/_inject.py,sha256=veCB5e29e3Axw2FJyv41FN6x1p26iOFSywuM7CzTUaM,10353
optax/schedules/_inject_test.py,sha256=8XQjwBKw9KQS-BXcjyOVwtV4G4SIVUI7E7MM1hP1Dt0,9558
optax/schedules/_join.py,sha256=k5UengpNrjiXPXa1SMSSp2uSUQYf80crjouWhjWFiU0,1644
optax/schedules/_join_test.py,sha256=NFjwadA4Udwd1bDmeJMBwa0tnG8bmuDiQsidehD6zLQ,1654
optax/schedules/_schedule.py,sha256=7bILgpes0cPtb3K1WKDOADf0erou20bcL6IjGbvA70E,24076
optax/schedules/_schedule_test.py,sha256=Fu_soXWIwquIzoIYeEB2M3KKVWSCLF_sJo-wrCZWyL8,21531
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=yKyvxeRydWJNbgsH1zLNID4c1BPin_4ah2zC9Suqkgs,2705
optax/transforms/__init__.py,sha256=fl3Oz3hZ8nxpAYi6ngOXy9mbGqnIktoi7Hg6s2_TP6A,3719
optax/transforms/_accumulation.py,sha256=P9yTka-FSMScax78Yfz1Aaa_5k47OjuJdlrAnn3KwPo,15246
optax/transforms/_accumulation_test.py,sha256=rlg_EjKgp7mg-AYZ2QX6Zk9wAY7-ItjC9xIhwClP3UM,14262
optax/transforms/_adding.py,sha256=LeHIp0B0gwpYvGMtvWc06WeTKvLspr9ghFYkHFXzyUU,3741
optax/transforms/_adding_test.py,sha256=1fFyU-KZkV4lN7wbx5vf7N5rDuTtvlRI-LlqeX8oI9g,3717
optax/transforms/_clipping.py,sha256=5Hyk6hID4y_E1l69NYAbT3GHqbQNBzTnhUs9DInyZWE,11305
optax/transforms/_clipping_test.py,sha256=FfSR4afxk74SkpvXQmUB4iGxVqA1bE6LLx9-RaJmjdQ,6946
optax/transforms/_combining.py,sha256=WPuVMq_5G5fziJM-irE2ZdV5CZza8R_3Lphp2wHwAFU,10182
optax/transforms/_combining_test.py,sha256=zDl7XXyl5bpZEN1kTi8R1Dfvc25mHtIE2AQrqqBnuJE,9355
optax/transforms/_conditionality.py,sha256=eLWPeMQECqasgNq1MSxJJTt6Vpb3siWjS147PViOqJQ,9435
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/_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=zdPLDp555QHyFbB7LcW8hqC5VMdlKkfJ--7v0jE6Jtc,13376
optax/tree_utils/__init__.py,sha256=kDO7csWe7bE-uownPnTP_XlYYBgbw8wiRzoWKkRCX9o,2407
optax/tree_utils/_casting.py,sha256=Dvm3zlFJHxDNzBWknx1kIlddTk7-0WmJTf2iKfJjMR4,7189
optax/tree_utils/_casting_test.py,sha256=svB-EUqmYwkX-WRF02Ag9USq3kE515zMHfs7dPzQ0Bo,3936
optax/tree_utils/_random.py,sha256=2XTK0flIDkYDzBlb3zmCgMf9MQzxsHb9wCtl_2aXi24,2562
optax/tree_utils/_random_test.py,sha256=XN9nThmQuqOE2nOdWE3liJaNKW7MCCN1wsNTVhfeo7c,3773
optax/tree_utils/_state_utils.py,sha256=t3iLZ5zlF2_smkLRCxglk3rKBrR7e_qRtR8XLB6Vvrk,27881
optax/tree_utils/_state_utils_test.py,sha256=yZUL5iqxPUxmd6-I6H3g8UeVJ95Yph-xN90eUG4g22c,20311
optax/tree_utils/_tree_math.py,sha256=WjP5seowDBDkHwWOrJ5Z7Nm8JvZ-woPbxtcTEAT1pBs,9579
optax/tree_utils/_tree_math_test.py,sha256=bxtJVmzzB9yHqRO96sSVosZO5MmTLp0gpLnkdkmx_lc,10129
optax-0.2.4.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
optax-0.2.4.dist-info/WHEEL,sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI,82
optax-0.2.4.dist-info/METADATA,sha256=ZhqPa-rAYHbgDd0FyMurdcRQCNIZfybZIu6K926l7QE,8283
optax-0.2.4.dist-info/RECORD,,
