dm_haiku-0.0.16.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
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=xXhakakByXfOT_2vSGpPYMLJ2HnqhNB1CNMOfjJxPt4,11073
examples/imagenet/train.py,sha256=yC7Ewn9GjcRDM3l1AG6Kc3hO2fmpRf2JhL0j624NS9w,13258
examples/impala/__init__.py,sha256=Hm11zlfCPGklSZW5XmFxd6x9zyrSqFPzHwF45CCt36o,696
examples/impala/actor.py,sha256=lPovBYiE8MAJHa1qiyrzUY6fh4VY6I7RXaKk61ZwxEk,3735
examples/impala/actor_test.py,sha256=J4LWvwJtJ-LmVivWE9HxLIj5L8oTilBPsaZ1UKenwIk,5627
examples/impala/agent.py,sha256=3qGVyh1O_JeBvUMe_4tso45j82Ct6PNBS4Xd9x5Or8U,4279
examples/impala/haiku_nets.py,sha256=f_1ntRQmNJIQ2fkJM1DY666Il9OBK6jJugcVLvm9M6k,5296
examples/impala/learner.py,sha256=u6LVVtiX3et6EBFRsOsSAyXa3Qmo8UlbeTL0mUdJEoM,7579
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=H3mhk6AO8eWIPhuj8zEMapZRWEIX6XYhqBSIWfhZ-8g,4538
examples/transformer/train.py,sha256=uG2lWf9hWnCOIe33bq6AufpY8Tf7tpbdbxxHAn07Xz0,5805
haiku/__init__.py,sha256=vx4dEQJX61d79sSKsc5BZT5nEVhdyWOpo3pyUX6soKM,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=sMXQAJUs0x_nsZ9tOGXbD0qGK2ModCmytIo3daBog3s,6801
haiku/_src/attention_test.py,sha256=MJktF_jJB2rv9FBeCEuRUPVm14y5CcXPy8ERpkZnaZU,5814
haiku/_src/base.py,sha256=BLof9zYyClstTgVm3A_iImOeeiUfUGAUSVndRY0J5Gw,47489
haiku/_src/base_test.py,sha256=86M8SCTb3ln3Uwbz_iizBqx0BEU9CQi0zjWt3mZc8Dw,32045
haiku/_src/basic.py,sha256=T2LIyWxbLwv9o6mD1fhp40_92rRiCx6qeowZENFCLJY,12448
haiku/_src/basic_test.py,sha256=D6HXM7ixOCTBIRs5rOKj_hN8pHjH33EeDeKfsdCiPx4,10304
haiku/_src/batch_norm.py,sha256=iRVLjXJ8zE2PD4_i87r7fbe5TizEibrnoSFT3Lz85cM,8557
haiku/_src/batch_norm_test.py,sha256=oHOLmY_r3hDiLqK3IahNYrN2al2ckPu3ROn04Vj-5po,8724
haiku/_src/bias.py,sha256=qWqnoqFmjw_lLlk7Ttqev4vbGIgFaeEDObiRSCjarHg,6055
haiku/_src/bias_test.py,sha256=_N61C6dj1OUwC0lNYKjA_0zdOsGIFq3w85wYoX7nS68,3213
haiku/_src/config.py,sha256=rABH5tTn-Gr9gLilbmpqxWyJY-KOhLG_fPosp5sB7bo,9549
haiku/_src/config_test.py,sha256=ANks6gWaolwaZBrfianZ2YLboif-o3HMTPJDWdbXwLU,4565
haiku/_src/conv.py,sha256=WkNCNXjTDuk9WH7togmXOTQOhdDDxhstjCZ3pY5-T4Y,31948
haiku/_src/conv_test.py,sha256=nh_rpqYnQA2yDJLBD6wr9Hg6cNu9yX_AVZ61g3c9Vv4,28866
haiku/_src/data_structures.py,sha256=XiSQVNmBKnGcecqb4Jw7801IQWi0-WpJl8bZC1gLnTU,10783
haiku/_src/data_structures_test.py,sha256=VlSk-_9mHEfAS3GbhaB_Yde-R3t5y9FRyEw3cOqVimY,13890
haiku/_src/deferred.py,sha256=IxeY2h3e82_OQNRxO2iLP_JFPf9OAF7yj38SZt1hkcw,5185
haiku/_src/deferred_test.py,sha256=fFgC5RiSCFwpliwC4Sh-HnXW2ikjfToYiRhF_5yFGKg,4646
haiku/_src/depthwise_conv.py,sha256=WRb4lyxH7dBqgcQXsQ7mlt-Ka7bSNLf2eKzXHRCEMTc,14373
haiku/_src/depthwise_conv_test.py,sha256=gqS1z6eBB84E0i9NOg_pt6dkB5Rp2PKFxdSxLv_hB7k,12738
haiku/_src/dot.py,sha256=6znnfAk0liLEjD0-X2F7Atf5HneCm9QRRGzZz5QN-Kw,13830
haiku/_src/dot_test.py,sha256=3lLPFbFvQxAat4hBw4ZTpXeHHbQGDEMjBh7CWP6y4K4,3710
haiku/_src/embed.py,sha256=CM1xstz3yFui9Yi_85IVWT57XZ_Bg-z0VNHoTXSpbAw,8030
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=yRF41TnsReYWWBMnkExVj4xN3aZVEWToiuEwgULbHyY,9118
haiku/_src/filtering_test.py,sha256=RHHKaxLlKDpBp5v0ADrduaAHjeS4QRZh6GrdXdujX5A,13597
haiku/_src/group_norm.py,sha256=0BCjYrFIj3sgFnXF39Dm1sy89Txx3Z6RkLKPBP4U9-E,8879
haiku/_src/group_norm_test.py,sha256=-C-sVYd4koesZJIiLlxY-HXkVHIrQPl999LdVINlE0k,10704
haiku/_src/initializers.py,sha256=br2x2_Y2NoCs5qZtnEyZ3qBHhHDqJvggK1TFWhhiEL8,12146
haiku/_src/initializers_test.py,sha256=lLyI0DceNqiPT1g7AzeniBjJAI5vyB77hcOAMD46mhs,7862
haiku/_src/jaxpr_info.py,sha256=lK-AqV_ZC15id3At1wiNM-4Y23YY4NIauTBkS5iz12E,22041
haiku/_src/jaxpr_info_test.py,sha256=J7M18tQzP1kaKVPDZSUb91ZVf6eDblzy-fB_vpGBKvM,5686
haiku/_src/layer_norm.py,sha256=Hf2UtEJeaAFKSCprZllUWOUNikoJpaTD8FlQP-AAjHk,9961
haiku/_src/layer_norm_test.py,sha256=UudSe8BKJzcYqWyj0qqF-rnAJEmZx_RWVaIPMn_Hx0M,9340
haiku/_src/layer_stack.py,sha256=fptIK18RV4pEI-_CMbzOUmaaxuUCjUXyRISasT84KHw,13754
haiku/_src/layer_stack_test.py,sha256=EML1UgUmizgBrGk6je_pAzv5rfbgNeqfSFlw4b01CZ8,23158
haiku/_src/lift.py,sha256=HCTV46Jd7oVagbL4Lx06-Ots7HiF_J2ppLdjUhkWSXc,18778
haiku/_src/lift_test.py,sha256=YUFXFeDJucDqTuCifCzTDMrv0r2GKCgQvwqoAL4ZXZs,17784
haiku/_src/mixed_precision.py,sha256=9iAxaz69uuQCNxRoMlvBnWg50hBA0SgA250JJCVyync,9603
haiku/_src/mixed_precision_test.py,sha256=hWaS8GLHYWprTmMAGPycVpjhAN2rdztpPWzwiMJvch0,8613
haiku/_src/module.py,sha256=Iqxeb3PtHJoyAb0KiQZVbKy_fTVd2NQSnPxxMjvP2hI,30099
haiku/_src/module_test.py,sha256=xwH5j6dqBj2O2urw8Xt9REbSYmvkKo61VuZwxRc6hLs,34600
haiku/_src/moving_averages.py,sha256=xmHIpji_jRlTIlbQxl6Sjg_gpM3jUyItUOlIdzhYj-A,7769
haiku/_src/moving_averages_test.py,sha256=8yHEp7LEQdxqulMcr5W2ZFA0tV_DQMcrQd3OR61vX2s,9477
haiku/_src/multi_transform.py,sha256=PikJedK0x2_BHHl263SXBZOcCXbPmHMP7XxzGxN1h14,12673
haiku/_src/multi_transform_test.py,sha256=vLdb3E7y-ry3CuRHfx_FAeJQSL36aCUyjHSTLKmQWMk,6169
haiku/_src/no_flax.py,sha256=5Ym4FiVndU3JEZnOwAe2Cd34GJhRXzJA5w2Iazv7wjs,1311
haiku/_src/no_flax_test.py,sha256=r_l70DbIG-1xUFEYl65eQwIzeZXLhClNPA0WU59_WSM,1844
haiku/_src/pad.py,sha256=0fb6cB5iKgitB-YaoMbC083RkhqLlMiH2d0iQ6X-dRM,4560
haiku/_src/pad_test.py,sha256=3VTLD2xvDfYpqXTk1ORqqqx3jCBUlRwA5O2iVdqk76w,2835
haiku/_src/pool.py,sha256=bFBEdzKV_wpmJJl3S4AtjfNNwJm7c9AD5tK0AzyPRow,7218
haiku/_src/pool_test.py,sha256=7uIL4qfCQubD-izD7Oy941RgH8r9l3r_jaMgukXHthQ,9267
haiku/_src/random.py,sha256=4tiyvgYjuLXxjhIZRoD_MnW5KF0ybp9G54CvW6jKHXk,3872
haiku/_src/random_test.py,sha256=qzoV4ykQ2-wy1anawHs8AGC5m6Q-B8MfTj04802sk9U,4076
haiku/_src/recurrent.py,sha256=y7-Jbbw9QS0AxBagBJjqJ4cVzqcbik6j3TWPyGUtB7g,28032
haiku/_src/recurrent_test.py,sha256=fwpaZBbZwgQgDwV4UNckCmTf4qvcEN4zs7A2JkCYkfM,19207
haiku/_src/reshape.py,sha256=H0D3nnlJamsuzETlTQL5NHtv31PKZG2j8B7YqHzhE-U,5581
haiku/_src/reshape_test.py,sha256=_z3hXWG8a0ndCSdho-ibn6JX29RQyVWeVI17Lsyhs_k,4754
haiku/_src/rms_norm.py,sha256=v4d_zl05O86BFqBRLTjTAkHcPdpL8LvCd1rZ3rWj_LY,4564
haiku/_src/rms_norm_test.py,sha256=5GFBqBp7zniqVqyxZBC57sZQ8Zi-PLXJ-QAuK3IucX0,5071
haiku/_src/spectral_norm.py,sha256=AOoUnKac7GKYrgwhqWoAQdYu3XOlrjussZ4wYciWg6I,6744
haiku/_src/spectral_norm_test.py,sha256=2JVl_1YsYUursw22FWYYr0WS51hZ-Wf--MeO_K1wvU4,3251
haiku/_src/stateful.py,sha256=G6vZ2Ut9zzEZV2VGCDWI3VdtlbuFJ_cDJTNbFEdWlmU,33453
haiku/_src/stateful_test.py,sha256=TEy__QubN6sX3KiKUNyLtlw2trRXDEzsvt9ZDKFxC5E,31343
haiku/_src/summarise.py,sha256=3a-P8Rb-GjrerQUvJifZsLSehju6bs60Lo7PE9jU6ic,15446
haiku/_src/summarise_test.py,sha256=TmzmqSkzGA4qgYWSDPtX_IorTMhO3nUEhCrUfq4puIA,8676
haiku/_src/test_utils.py,sha256=XW7Relj-H66XklDZVuOwAATC6HYMJ6xqxCp7ssulHTI,8052
haiku/_src/test_utils_test.py,sha256=eFZ1taKHCrex7HWUaCWuRdTgrq34u31pRXRauILRH_M,1347
haiku/_src/transform.py,sha256=3a_arWa3BH2NEjvaj_CzB-4PJPvPs59IiVh32btVU_A,19280
haiku/_src/transform_test.py,sha256=w4hHt_bH-UARm3gJcvuJmN8zWJpIOmmNWa33ENLstCg,22691
haiku/_src/typing.py,sha256=SZq_A95eViLQmyJtFF4wPxDcG52GTlPjssPaLtaZnb8,3041
haiku/_src/typing_test.py,sha256=4NjsE1TCU9YbwR7IWIgofNdHrecvUFpy9WUtD1Dop4M,2297
haiku/_src/utils.py,sha256=jdpiOW2vAo4WxV04DKuHL9YY1G1vShIfTE1riBmze90,9103
haiku/_src/utils_test.py,sha256=nC0hl9uW3p61NZraagwY6CT-Py9W1wZRrUhm6ifROyM,5779
haiku/_src/flax/flax_module.py,sha256=_XX10_ZC5xG4rcTI3XAdiLAcACNabg3gb3SUVpGMAmo,5979
haiku/_src/flax/flax_module_test.py,sha256=d_Hunbrw1WcmDjXy-rhEIVBNYFGFBK1OqjiD_G_I3vM,6216
haiku/_src/flax/transform_flax.py,sha256=4hc0uli2Ln6UqU47BMeymmkDqgcygMdnCWRb1-Pa8gU,7336
haiku/_src/flax/transform_flax_test.py,sha256=0eJP9Sgr5Eay7cAT-tgkh0M8_mP8xZ2uDiQD7MmMlDQ,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=H-sTg4-vLRDDbwlSYELmFkvgFrjghPhmfHJvxYwDtBE,2105
haiku/_src/integration/common.py,sha256=WSgD_TKCLJ8CpbFzRzZ6lhIMuxiHlcF5JcLf8twoIhU,2808
haiku/_src/integration/descriptors.py,sha256=lPmcWR1QAvW9fnQMZoFSgya4k-YmoOpfU4iuY_beQmo,12746
haiku/_src/integration/descriptors_test.py,sha256=nkSyfJmgFGVOFDGk_f1B9uNUBl26WUvPh5VeAGDq7vM,1342
haiku/_src/integration/doctest_test.py,sha256=458pSq3SpxWY-HwYB0imtPffn5o0Xi-2quYmODS9j9c,4337
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=08fg8ab78d3l6qTNf8xNO1LgPvXENPg2wNIlHpfBJ_g,8671
haiku/_src/integration/jax2tf_test.py,sha256=IKP42TS5Lkty8ruRcF4cdvJ8e9-3uxm1khuQtBqyTJ8,2814
haiku/_src/integration/jax_transforms_test.py,sha256=IjBkJvXHtoixWEluldr5A45S1NlHrOJF-H2dtARxjiU,3144
haiku/_src/integration/jaxpr_info_test.py,sha256=gmhPHjopxocwXDOG5YPiz4Gz0QiLbavnwasHctIbcQ0,1647
haiku/_src/integration/numpy_inputs_test.py,sha256=h4RDh7xPyFXjYSnCOFWkA1kc2JAQjec2t7Jb12lXHRc,3388
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=FAu2Ertc-YE8TxxoivY0dLF5KE-mYLsj2_rzAz2jjYc,6036
haiku/_src/nets/mlp_test.py,sha256=EI-KWD0zSSSduY2UJnpmjXGlgZ4GCaJouGVIxyyZjKY,6918
haiku/_src/nets/mobilenetv1.py,sha256=m9W1t8YavqCaFQcbFGsJhkNX4CUNED0f38tUxsbf2jg,5096
haiku/_src/nets/mobilenetv1_test.py,sha256=BeyPKPo3VXou07MWvnYmAyKnxlwEhOsBx-LAqHyeHUU,1912
haiku/_src/nets/resnet.py,sha256=40UaLAbafalwJr1L0lnagQNJ4uDgZmXvsIBOPc0Y_eg,21648
haiku/_src/nets/resnet_test.py,sha256=M98Gbiv5iDC8E0laQad6OElK61QE60Rj3t3KEYPnARk,6687
haiku/_src/nets/vqvae.py,sha256=vYehRMADzjgNvtCVfigUtrP94Q5UlWLY2cKTkaRDC2Q,14863
haiku/_src/nets/vqvae_test.py,sha256=vNcWZPxxF92kiDDLIVCCPB8da9lqXplLBYVE_ZivCzM,8020
haiku/benchmarks/eval_shape.py,sha256=W2yo5EWG4HmeCNopGLTDgDmwd0IBTtLZ-rOmqLJAzTc,1900
haiku/benchmarks/init.py,sha256=oUM8bNp-JoucsA_HsGHP-7eLu13pjJydey2TyZc3cpY,2404
haiku/experimental/__init__.py,sha256=ZmO5w0y4-OI4WR-9WtfPsFNWTvRXm4HQ6oJia92MaaE,3700
haiku/experimental/flax.py,sha256=cAko8LBODdwR0Gx6jkc4Zvh3nBby8zwqXGrj78FH1s4,1000
haiku/experimental/jaxpr_info.py,sha256=mi9fhHdCWnDViKB_phhGOass_W-gf-qlKwq6g2zZDp8,1281
dm_haiku-0.0.16.dist-info/METADATA,sha256=5RaH7DN10hBFbdUkPQJpt0WhYSH-9fIYTBQ-iTy7MDk,19841
dm_haiku-0.0.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
dm_haiku-0.0.16.dist-info/top_level.txt,sha256=IpsXnvwDmYOADfkGYRyCOMyRwJOk6dTsVB-X_lxoPjM,20
dm_haiku-0.0.16.dist-info/RECORD,,
