docs/conf.py,sha256=oVLTco20uBxHGNZa34BK2O1QnZX2NW58I2zo0N45vBw,7757
docs/ext/coverage_check.py,sha256=dO_6_qqwlEWszmVXRA3cCFORiqcFogy29Xd8UqXZdLQ,2772
examples/imagenet/__init__.py,sha256=Hm11zlfCPGklSZW5XmFxd6x9zyrSqFPzHwF45CCt36o,696
examples/imagenet/dataset.py,sha256=rHErhCAQtHpOn0NZDe2Mb7mXznBGzDuwcpurdFtT264,11120
examples/imagenet/train.py,sha256=7_VcztsrDCAXIhJU_UpStVYxMutTgsHymrtShil8Iuo,13269
examples/impala/__init__.py,sha256=Hm11zlfCPGklSZW5XmFxd6x9zyrSqFPzHwF45CCt36o,696
examples/impala/actor.py,sha256=bSSsS5TI0jly7IcfloQeWlYHz4JN0mAoCTlF58_lyBA,3745
examples/impala/actor_test.py,sha256=J4LWvwJtJ-LmVivWE9HxLIj5L8oTilBPsaZ1UKenwIk,5627
examples/impala/agent.py,sha256=gM_z63mueYcNOommV5Uu5lzyrVZWMrXVTHbgV0R4lps,4312
examples/impala/haiku_nets.py,sha256=6CKEOJOFht-TVjiORDS_ZdHvt9-bzT-kz7c9hs3VK94,5306
examples/impala/learner.py,sha256=yqrbjgcpV49XbUXdMXixomz76cDYNYGrQ9veZWMZQR4,7649
examples/impala/learner_test.py,sha256=3f9ufdPyAH5OeMZisf1Plw-RL2lCimKkmiCVzngxsc0,1943
examples/impala/run_catch.py,sha256=VgfCzThcE2qcfESE5KWcfIQjdK_zQDug9zHzAOXJ_TA,2979
examples/impala/util.py,sha256=kJ2g_j5-iHKLGwV3YAJO9neGsMOmq9aafaLpnRJEBK4,1589
examples/rnn/__init__.py,sha256=Hm11zlfCPGklSZW5XmFxd6x9zyrSqFPzHwF45CCt36o,696
examples/rnn/dataset.py,sha256=9W_my19J0EojVHY2B09Vy8-JUxfvPEueuoWEJy0QNwA,1890
examples/rnn/train.py,sha256=gXFVe4I-U8QIFhEmgsI33Yksi-iSmpD3efZFoPJ7liw,6593
examples/transformer/__init__.py,sha256=Hm11zlfCPGklSZW5XmFxd6x9zyrSqFPzHwF45CCt36o,696
examples/transformer/dataset.py,sha256=5d9hZkG6cpY3Ao7Z79eEVqq28Slb7wVxKiwowwVasRo,2735
examples/transformer/model.py,sha256=4C_UYBK5-0l1iBjfNsmzgNhBoVJ3hhMq5wRRujRN8Z8,4605
examples/transformer/train.py,sha256=HX4_n9MzuFp37spM49WyAkM-4SNBCLUAl-At6Y_dMY0,5818
haiku/__init__.py,sha256=I7-LIU3zmE_lAXjTVYKFsSBjputnGZcfQd4xOnrOohg,9243
haiku/config.py,sha256=0qHSqiDlQrZ1hRcJG9dgY992RLUvTqfM7uo0YKISE1U,899
haiku/data_structures.py,sha256=Dw3W5pyQFXrKyYA0O-P3p7kvUVbkPWfAxVQytoMFWM4,1604
haiku/initializers.py,sha256=6ipg9-VBFaGQGDA54t_VoMeh59rCmaf8woz_0PNqWik,1410
haiku/mixed_precision.py,sha256=5_XV8OIxx3F9XWVNJy3c472dwJjaG9MYeCKbRepEA7g,1140
haiku/nets.py,sha256=-yqFSayBW-25graY8H6r_IXXYFdLdJy1AVxlkjGKkGc,1490
haiku/pad.py,sha256=yrr8h3UMaIEFEjN6BItabrLyVkfRMQNUXdmAUpfZ730,1308
haiku/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
haiku/testing.py,sha256=nv98zWFoG00QyNXufYtLL7wKOeE7Mo96RqJkXZHG6hg,858
haiku/typing.py,sha256=ocBOb2VHYBLdnaE_l2to1LHPRczVOavgowyK-vpp1tg,995
haiku/_src/__init__.py,sha256=Hm11zlfCPGklSZW5XmFxd6x9zyrSqFPzHwF45CCt36o,696
haiku/_src/analytics.py,sha256=jBW6hBmFtYSelsvkCewVAKMmy9PO0LT2Y8xv2j6-8r0,843
haiku/_src/attention.py,sha256=7PO6gljC13vodg2LE1zg9Nnhbk2162cSnUFQ86YeJIA,6854
haiku/_src/attention_test.py,sha256=MJktF_jJB2rv9FBeCEuRUPVm14y5CcXPy8ERpkZnaZU,5814
haiku/_src/base.py,sha256=nFFeOo1lslSQ_KfPQCfHmn3rCP1KHR3Y5VuK51nIJqQ,47435
haiku/_src/base_test.py,sha256=Ls_EGvVgK9Dklv14NErVlFa-EBo9yAwQZ6kTKfusO9Y,31837
haiku/_src/basic.py,sha256=cISYWBAChiyi1Gob3eqeM6r4cyuvZPZHdOYp2R4SKWs,12574
haiku/_src/basic_test.py,sha256=D6HXM7ixOCTBIRs5rOKj_hN8pHjH33EeDeKfsdCiPx4,10304
haiku/_src/batch_norm.py,sha256=qu7D7ycVWbzpe0IQr-ZBINHE0GW_06P8tWE_PGNWFO0,8622
haiku/_src/batch_norm_test.py,sha256=YcR1JVVxskkzv_I15uP4BRZj4-yeCO6ilwN8maesd9o,8724
haiku/_src/bias.py,sha256=HMYEvbwxIlbXsEU8oNdAZclPtu12taIk_B4yHOwtx0U,6111
haiku/_src/bias_test.py,sha256=_N61C6dj1OUwC0lNYKjA_0zdOsGIFq3w85wYoX7nS68,3213
haiku/_src/config.py,sha256=dtLar8mUzsnA9xmleTb7kM-tnE_u3PkfgRmjJiC1dJ0,9601
haiku/_src/config_test.py,sha256=ANks6gWaolwaZBrfianZ2YLboif-o3HMTPJDWdbXwLU,4565
haiku/_src/conv.py,sha256=K92oFIRM77tM5X1Vhiclfa2rrR-u-CTujE4PZFtsBwE,32153
haiku/_src/conv_test.py,sha256=nh_rpqYnQA2yDJLBD6wr9Hg6cNu9yX_AVZ61g3c9Vv4,28866
haiku/_src/data_structures.py,sha256=B20kynuytK1d67amyshiDr7_KzSZVMue2bJm_xdrvOc,10894
haiku/_src/data_structures_test.py,sha256=m3i_Qhf5z2m0Vy8Sr5th9OdhWO5VqedvKnOQpD2LvkU,14051
haiku/_src/deferred.py,sha256=lUrnoKbu2VpIki-R5chA1c5sRT1Z0PYLF9OaY9wNxkA,5186
haiku/_src/deferred_test.py,sha256=fFgC5RiSCFwpliwC4Sh-HnXW2ikjfToYiRhF_5yFGKg,4646
haiku/_src/depthwise_conv.py,sha256=zvnDY13AfBVRS2VsRUtDgwI-2RjTQuZ6ddjBD7c7mjk,14567
haiku/_src/depthwise_conv_test.py,sha256=gqS1z6eBB84E0i9NOg_pt6dkB5Rp2PKFxdSxLv_hB7k,12738
haiku/_src/dot.py,sha256=xnd3lfb6xbEIVXVNGWwOZVddXQ6cfCP6PkkiyRFflaY,13425
haiku/_src/dot_test.py,sha256=hA7k3kqLzbq0bVojZlLgylbfokKt_f3iVCLMoNC_364,3390
haiku/_src/embed.py,sha256=4WcPLQ58dXA8oz4rK-S_n_C4Idux8c17IznID7yORUI,8110
haiku/_src/embed_test.py,sha256=etC9k5Putg8t3twDtoNY5Ce3f9QjJpNMeGTYBN3Vy0g,3767
haiku/_src/eval_shape.py,sha256=zzzZuFBdU-qqOSkngpktCBhCkuMUneocP3-JH6avGmQ,2151
haiku/_src/eval_shape_test.py,sha256=HCMqMV7W-0A9CSEkLL0sxwaK5opwj919O1muoYXAjk4,2981
haiku/_src/filtering.py,sha256=3nB4lf2voPw3ZfKy9eC87j80l0BqjZB11okANf-e69g,9120
haiku/_src/filtering_test.py,sha256=eYqxKbMoubi0bcYlaDI81G8I_A9Y0DG8yBWyLYlIY-k,13607
haiku/_src/group_norm.py,sha256=6cLIkJebZiLDapUIa18scoarFqrBdebZPaFJyOrfevQ,8934
haiku/_src/group_norm_test.py,sha256=-C-sVYd4koesZJIiLlxY-HXkVHIrQPl999LdVINlE0k,10704
haiku/_src/initializers.py,sha256=0DPi_dUn6tNXbsXiEB2JwNqEZsT6xHDJuSSstrkfkhU,12182
haiku/_src/initializers_test.py,sha256=lLyI0DceNqiPT1g7AzeniBjJAI5vyB77hcOAMD46mhs,7862
haiku/_src/jaxpr_info.py,sha256=QNEHzLwmkGqBa2hKY7OnvPCir3OhcQuElBCsxMF2LTc,21649
haiku/_src/jaxpr_info_test.py,sha256=TIdzj1qHWiOcV19Dna6uLvNps4quZwf8Mjp-vR2ZNzk,5021
haiku/_src/layer_norm.py,sha256=UTHMA-4N1LCOkD9W86NiYkfpgUguks2AK0YRI4Mli9c,10034
haiku/_src/layer_norm_test.py,sha256=rZYpz6pKGokUomMvKjZ3AriPxH31VEp0hSZQiRsN3ks,9359
haiku/_src/layer_stack.py,sha256=edKn2G3q8HD1IJ7dlz0d0y7RWRqQ6HTxhVQZK067rzA,13771
haiku/_src/layer_stack_test.py,sha256=6r7AWvDAnbJs_WNc81z3HbP7kfnTToGT4Osyhd1Lu4o,23173
haiku/_src/lift.py,sha256=rI3-z7pJdq4foRTWkVthz6iZ9raAvxtlbIO9_LvBI9A,18801
haiku/_src/lift_test.py,sha256=Ck29PbHKNtNjNAtSgCDAc2U_dJm-KG2BKMH7kWxUYSI,17946
haiku/_src/mixed_precision.py,sha256=b179iVJPO1m6XaaKvU2y9mGhIvzYadMUNsaRdsnWVpg,9622
haiku/_src/mixed_precision_test.py,sha256=UHAW2UsN5Iaj3eECPd0dJlIqG0ymxrZY0elp2HDqfaE,8671
haiku/_src/module.py,sha256=CF_w7TZQKWXquF2LHoElE7pYhiM8wpWsRPZ9KrPCG7k,30135
haiku/_src/module_test.py,sha256=HD9u4fBl-1QwF4cX4a4xKmn7cgcqLep_VwfFZ44dR9c,34606
haiku/_src/moving_averages.py,sha256=SnDY4rgZMJWXRQeO904NgcTEL2dZxQ_8xk8IAEFr67M,7816
haiku/_src/moving_averages_test.py,sha256=8yHEp7LEQdxqulMcr5W2ZFA0tV_DQMcrQd3OR61vX2s,9477
haiku/_src/multi_transform.py,sha256=aqJHBVe1NUZhkNYrC4OJ03FuJ-phwE65KnabqYYuYPM,12716
haiku/_src/multi_transform_test.py,sha256=XVLlSKwLKYe35vEuurp1HRqWIJTcYFPBFEQSP1IHL_o,6230
haiku/_src/pad.py,sha256=fL4s49JxSHW0EOkKdlI1xIGV11OymvpoQOh3ZQfzdS4,4528
haiku/_src/pad_test.py,sha256=3VTLD2xvDfYpqXTk1ORqqqx3jCBUlRwA5O2iVdqk76w,2835
haiku/_src/pool.py,sha256=r94Z3M0_gFHF5-1EfIFwE4tafZ9pepqoY9EkhPRvFRA,7328
haiku/_src/pool_test.py,sha256=7uIL4qfCQubD-izD7Oy941RgH8r9l3r_jaMgukXHthQ,9267
haiku/_src/random.py,sha256=4tiyvgYjuLXxjhIZRoD_MnW5KF0ybp9G54CvW6jKHXk,3872
haiku/_src/random_test.py,sha256=QrXGGiBQVPFXC8FldL5uwuodJbtYyqAAkT4VkePBPDY,4110
haiku/_src/recurrent.py,sha256=VlN9cxtGhtbAv969cwbmiwXt5Gis4bS1bnLjzMxHQZI,28150
haiku/_src/recurrent_test.py,sha256=Rmxb2jka7LQnPHi6PvoPhhT__hM6Kjt1F7tDQbWfoIk,19194
haiku/_src/reshape.py,sha256=hu85XRgy3yLimOVaRdhiqt3NnG-_oE9sW95W2GTJ9uc,5615
haiku/_src/reshape_test.py,sha256=kUQSjYgNWfRRxC_jwu7omieXKcdRnEuADf7vI2DiptA,4774
haiku/_src/rms_norm.py,sha256=QilM2tLrEG4pIZV_PHy9ggpMifghQTN7Eij9aqZVZzQ,4583
haiku/_src/rms_norm_test.py,sha256=BBse-AM0bYQr2zf8f1z7FElYJXrwkHPveRmWijMrJoc,5090
haiku/_src/spectral_norm.py,sha256=0fcsgEkOjo2A0IeiJibR24cIPkNKAgTEqDh36srT6Os,6778
haiku/_src/spectral_norm_test.py,sha256=2JVl_1YsYUursw22FWYYr0WS51hZ-Wf--MeO_K1wvU4,3251
haiku/_src/stateful.py,sha256=1qgLlxnOXZAnlhmvbPGVb6OD2a5ETEIWohB3k3u2Cr0,33561
haiku/_src/stateful_test.py,sha256=Ri4RO0LZ6_l0snac_jV5Oj7NvLwQI2vYjj2kFUUPwuI,30654
haiku/_src/summarise.py,sha256=9E5UpWX1TsOfTSJcuWEST72Nb4fNa73DYmD1sG8tA1Y,15512
haiku/_src/summarise_test.py,sha256=cIHV96Ed_hNhrkoxhdPACtgdKeCWf5WOmBxADx6JV4g,8686
haiku/_src/test_utils.py,sha256=lOdqbuBjflc4Q6439NJssXqq5BJSO6bxSmYL8Tq7eso,7909
haiku/_src/test_utils_test.py,sha256=eFZ1taKHCrex7HWUaCWuRdTgrq34u31pRXRauILRH_M,1347
haiku/_src/transform.py,sha256=7BzopLpUGBHZ-B60_v8D4QNw6rXt48cuoqXi-dvc_sA,19437
haiku/_src/transform_test.py,sha256=56YhSUfVbzrVJrSwM9_zRemWUKSjvbBdUa_T3VwM9TM,22532
haiku/_src/typing.py,sha256=Ssmr7P8kc--QZrc-QRoGmNnouq0Q41eDtbr5L0Fpx64,3041
haiku/_src/typing_test.py,sha256=4NjsE1TCU9YbwR7IWIgofNdHrecvUFpy9WUtD1Dop4M,2297
haiku/_src/utils.py,sha256=uIcv_nuWWFbAipCXzHGQa1gDqZFZJl1L5_SOBlI9AVk,9156
haiku/_src/utils_test.py,sha256=nC0hl9uW3p61NZraagwY6CT-Py9W1wZRrUhm6ifROyM,5779
haiku/_src/flax/flax_module.py,sha256=mZLcLQTHgGym8FWToUflNw1JSgWjEuigwWqZW213U6E,5992
haiku/_src/flax/flax_module_test.py,sha256=d_Hunbrw1WcmDjXy-rhEIVBNYFGFBK1OqjiD_G_I3vM,6216
haiku/_src/flax/transform_flax.py,sha256=mFj2x7u7ZIAtdLVtgU36MecChnCA5UyEMEuQC68_hxs,7336
haiku/_src/flax/transform_flax_test.py,sha256=HISf73tq8UxJsXDX_IRaWIiQJR6ch4K38AcYrvNsREI,7947
haiku/_src/flax/utils.py,sha256=RvrRbBjeQsFRpOLQoOwcTu-KgOr11lU-FWfud7eEYEU,2327
haiku/_src/flax/utils_test.py,sha256=y_3zq2uzdwn8oq0qJoLlTCv2F86N6OgQBdTmp19lNlk,1385
haiku/_src/integration/__init__.py,sha256=Hm11zlfCPGklSZW5XmFxd6x9zyrSqFPzHwF45CCt36o,696
haiku/_src/integration/check_tracer_leaks_test.py,sha256=foMzlCfcMkl4UHPvPROgFZzs-SBsupR2GXwVNWQgqeE,1847
haiku/_src/integration/checkpoint_generate.py,sha256=QvvgHbuj3YgrSCRhBoS9nAbBTfZaTSDNDgKk4c5mRmo,1519
haiku/_src/integration/checkpoint_test.py,sha256=212hWlUkyUb5Zn1-W8zf3imt2AcVRcaOtaM2SYc07WE,2083
haiku/_src/integration/checkpoint_utils.py,sha256=ZOFlwihKw_8DfuveYbxJfYXqJpj70BdCyRem1RHsZDk,2125
haiku/_src/integration/common.py,sha256=O3eUStIV4GntaMvFGfgETtGavYNOnZI8UoYcRLkT3CM,2818
haiku/_src/integration/descriptors.py,sha256=0DWjpGEvBhWevQN0spHd-fjsbobYUfyhAWUX_qC_JB0,12750
haiku/_src/integration/descriptors_test.py,sha256=nkSyfJmgFGVOFDGk_f1B9uNUBl26WUvPh5VeAGDq7vM,1342
haiku/_src/integration/doctest_test.py,sha256=od5v5Okb2dmgBYilo8pAw5xD0gUfMZU2vEouf5cx0_Y,4280
haiku/_src/integration/float64_test.py,sha256=ITMRGM2UNeqqafmcvdWkabWl2Rkjrs31qwZ-RzhBens,1837
haiku/_src/integration/half_test.py,sha256=7aPq6-aV_lM15nB7iAALglLxpfkoMhTHUENwzphGORg,1446
haiku/_src/integration/hk_transforms_test.py,sha256=g5iHf6ezECwoKMoBzHhgzyRHTFgX4vYW-hUn5tmdObc,8720
haiku/_src/integration/jax2tf_test.py,sha256=ZkqFMlnpNKZsxxMlXZvxvROPN5jqbY8tX6h0eJRmWhs,2798
haiku/_src/integration/jax_transforms_test.py,sha256=1gTRiX0hk7BSrEIGHrWpFlrBL3lx7oXWaBBfkDTNwAs,3169
haiku/_src/integration/jaxpr_info_test.py,sha256=gmhPHjopxocwXDOG5YPiz4Gz0QiLbavnwasHctIbcQ0,1647
haiku/_src/integration/numpy_inputs_test.py,sha256=536vHtsxTpY4EcSVjxGykyq8os1aFDjf8s7Vg7H2Moc,3405
haiku/_src/integration/rank_promotion_test.py,sha256=fjsUayO7idcH-T2eosFreotmDXFm-mJ-V7-JhaoXjwY,1967
haiku/_src/integration/shim_hk_test.py,sha256=zHZiLsiQTvNeVtrVo3V_RicsNVl0CsX1QcUmxXFk7lQ,2023
haiku/_src/integration/summarise_test.py,sha256=MlS2uvSNzW_h4qlKdJDcQueSikjJts7jhJ5c-vrs5xU,1649
haiku/_src/integration/to_dot_test.py,sha256=wvz90xx7_xOc95c8Ilm64PJZtZlYLspqDgSh_YtMLik,1508
haiku/_src/integration/typing_test.py,sha256=jWmj6D_Kui1R_w6aqD-iYGGY9XpdwTfEd2mD3XIDoSs,1426
haiku/_src/nets/__init__.py,sha256=Hm11zlfCPGklSZW5XmFxd6x9zyrSqFPzHwF45CCt36o,696
haiku/_src/nets/mlp.py,sha256=C8lAcYPipu-KA7qLxRCblmFel8Ab5GixMMneJr7msBk,6082
haiku/_src/nets/mlp_test.py,sha256=EI-KWD0zSSSduY2UJnpmjXGlgZ4GCaJouGVIxyyZjKY,6918
haiku/_src/nets/mobilenetv1.py,sha256=58Wl6uKWVwg6TFQZ8hotaUXQYrkrciZVxyLChzEiGvg,5130
haiku/_src/nets/mobilenetv1_test.py,sha256=BeyPKPo3VXou07MWvnYmAyKnxlwEhOsBx-LAqHyeHUU,1912
haiku/_src/nets/resnet.py,sha256=BZ8cJ4RvMJOk1xMADLIrexw_iZrFpPmPxxsTcu1LTJg,21769
haiku/_src/nets/resnet_test.py,sha256=CGaRuCuGSakkV4gOAnjgjWMxhsAIiUT4_kr1ExfNA5U,6706
haiku/_src/nets/vqvae.py,sha256=rMDInfdPJLQD6UManwsL1fwXimLr_JVM1Avp_2HUP5U,14886
haiku/_src/nets/vqvae_test.py,sha256=3X6KWMEPhv_3pLW41Ccj6JX3FV1QA_MrggRafvnLMwg,8030
haiku/benchmarks/eval_shape.py,sha256=W2yo5EWG4HmeCNopGLTDgDmwd0IBTtLZ-rOmqLJAzTc,1900
haiku/benchmarks/init.py,sha256=FtQns4wRSsqb2YRbgXd_kadTcTfzVK2pHHiYcdLTLow,2414
haiku/experimental/__init__.py,sha256=T0_ZcGhj4093mCOeV-b0-cGOoTi6dQMWBOf5SxmYo1s,3363
haiku/experimental/flax.py,sha256=cAko8LBODdwR0Gx6jkc4Zvh3nBby8zwqXGrj78FH1s4,1000
haiku/experimental/jaxpr_info.py,sha256=mi9fhHdCWnDViKB_phhGOass_W-gf-qlKwq6g2zZDp8,1281
dm_haiku-0.0.12.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
dm_haiku-0.0.12.dist-info/METADATA,sha256=IpvvmGHAPxJk7XeoZu7JzZAjBTQutf_0jjSPg1zpQu0,19610
dm_haiku-0.0.12.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
dm_haiku-0.0.12.dist-info/top_level.txt,sha256=IpsXnvwDmYOADfkGYRyCOMyRwJOk6dTsVB-X_lxoPjM,20
dm_haiku-0.0.12.dist-info/RECORD,,
