flax_extra/__init__.py,sha256=R53PG61WeA5QRgOhLPcZgZIsXLVQNqsj-oGReU7D0gY,361
flax_extra/batch.py,sha256=jwRO0v2NSpzHBpNh02julzvELVjLUhEM9bV1PiLFBHI,1957
flax_extra/console.py,sha256=H3YLRzU0JkehEwEwMFmec29xgc-nFbXp3sqfV9rJ0s8,600
flax_extra/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax_extra/random.py,sha256=9iE_HetVli8jzRQbG-m1jaI-mZ5nbR4fLIXY79nKylk,2078
flax_extra/util.py,sha256=u_luWmLqRHku41tOtAjX5ZsrTs883INr5LMskRM48HE,1268
flax_extra/version.py,sha256=qYq19k3S5-8F8MG3D8QOmJIWydzKll7VPSwzs8H3OWs,51
flax_extra/checkpoint/__init__.py,sha256=7U1ClJ3U_3rXfn_TXr8mFo_qjMNpzVIN2xs-32KpB4Y,1113
flax_extra/checkpoint/_best_checkpoint_file_reader.py,sha256=3pooVCAPAHhKezjtuH89VW6BcXcqwa7WMs9dPmIh5dU,2237
flax_extra/checkpoint/_best_checkpoint_file_writer.py,sha256=lBJeeC4L7pt6Kv4ytkWJNlNYJpjmG7MebsP3hQAA3Pw,6795
flax_extra/checkpoint/_checkpoint.py,sha256=0ZMYJ5Dtfn00K7M09nhGKIFP-VPfCdpyh9hgaZTt9G8,1002
flax_extra/checkpoint/_checkpoint_file.py,sha256=So3kT5xP8W5udPiML-dVh4riAlYvTtWxgEBl6v76Q8o,1283
flax_extra/checkpoint/_checkpoint_file_reader.py,sha256=CfP1jZtKQiwZVal9RifI_6Wg8RBlY0zuVhtpOjhFUfg,2698
flax_extra/checkpoint/_checkpoint_file_writer.py,sha256=ft-pG5214XuTAo1UgE_I1QV5uJfmr-QXF5UqJaVkQF8,2821
flax_extra/checkpoint/_summary.py,sha256=qmWE08KUAD0Njf4P7xed2ElURcqf4KUsQAUyuDcHqNk,395
flax_extra/checkpoint/_summary_logger.py,sha256=iEMyLSkFPD8G2j6qGaK2pEddYbXrposOd2lQcHL6mQs,1678
flax_extra/checkpoint/_summary_writer.py,sha256=pl6YqPcnA11GRMH6VJQbCLGb_e3uAbx_czllDcCu4eA,873
flax_extra/combinator/__init__.py,sha256=cmdx1SobNVIViKtXLihcez70KJNiLfCSq9ghF_FH0zI,519
flax_extra/combinator/_concatenate.py,sha256=J68Od1OK-ypMYfxArZrlQ1I9VRVFaS0FY_Gl-Hrb9dQ,1322
flax_extra/data/__init__.py,sha256=NThypTSjyIYwUGDc6PtAnvQgazAFo2R9m_CRKKvty9w,184
flax_extra/data/_bytes_tokenizer.py,sha256=D8lbWooKt9F5ChbdfIKMAvboBJlKc26RZwO6Th8VjHE,2872
flax_extra/evaluation/__init__.py,sha256=nfj5Pb-XT2TtWwFAcQvTFAG_FaeIV9CackhITjB8Itg,130
flax_extra/evaluation/_loop.py,sha256=QToWNQm2FyGmf6i5Xqebia_W1sTIz6HOFhBQ83QVbbA,7720
flax_extra/layer/__init__.py,sha256=l_suyw0vNBMekIcEJckX6VV_0t1QhInhkM0AfVa9_-o,2026
flax_extra/layer/_attention.py,sha256=A-_xnhd_Q45Wley-Dy2RoobP0xgTdsSTbhXP_2GBSxw,16876
flax_extra/layer/_decoding.py,sha256=vmEndQP1gMt1ccIzzGTT06gFjBGWkCjaPpVWAvbif_w,2049
flax_extra/layer/_embedding_decoding.py,sha256=Bkrb6qxF7F8n2CmDaYaPgeZv99F8m0de-b9Q5SVLYXM,1474
flax_extra/layer/_encoding.py,sha256=_3EFqPuWZr9dkKKpCNwt1Jcl8vO7P90Za3lKWRK9b4s,3553
flax_extra/layer/_feedforward.py,sha256=Dl8FF-BdrEx8e2U4jfSAcBB89wZjYL5ckRxbi3SikJM,2082
flax_extra/layer/_fourier_positional_encoding.py,sha256=lERaQV0URN5doe_0GroaeOMSC8wTjMRp_DaBqSLC1cw,8207
flax_extra/layer/_lstm.py,sha256=I5TIiy6oyDpU-VcCqaAPlMNKg33NC6k01WX7BDqWRe0,1592
flax_extra/layer/_multimodal_decoding.py,sha256=5XT1bODapu_50SLDydoBV5Zhq2lwK70ssEaE97Xoh10,3651
flax_extra/layer/_multimodal_encoding.py,sha256=RQlKcari8KcK5VvfjMSNnDB59CKSawMeeb3-_oa8BHY,6252
flax_extra/layer/_multimodal_positional_encoding.py,sha256=yF56Cc7q77zButg2_RYyPbzE-E-6MU7Pa7y6pJ1djac,5572
flax_extra/layer/_trainable_positional_encoding.py,sha256=thsCdYcYusjsTg7jgFmnmqlerwSukadrS7YtaGPihhM,2617
flax_extra/layer/_trainable_positional_masking.py,sha256=Mao-144E7k4EwvF5VcLlcJlPNvGc-0oZh5ktPxNBG6I,1569
flax_extra/layer/_trainable_positional_padding.py,sha256=6HHKtfgA35bM27kyIrWKOylJDduUGd-9lBBgGG1AlYY,1774
flax_extra/layer/io.py,sha256=ex1SqRgoMFkQUp_t6EfNzOeo3I2qZfr8eoYW8CVLYAM,2033
flax_extra/model/__init__.py,sha256=kE4_S_9YBFCWRIUcXRIJvuLUc0ckG2oEIlS6uy7tr_Q,166
flax_extra/model/perceiver/__init__.py,sha256=nZMgNcIBt_gc1ku4TUXHCNuoCdI0aNGHsEjL1G3L6zE,477
flax_extra/model/perceiver/_cross_attention_block.py,sha256=I6zohp6XhF0qYh_gGmxwXGXWbYNc5Owa0LQisqE7YqI,3463
flax_extra/model/perceiver/_decoder.py,sha256=3xmROYUudmUNts1juS5kCmW-6GFh-1tMUWkxffAs6qc,2925
flax_extra/model/perceiver/_encoder.py,sha256=-Ix14kvXfVE5AW9TPgg7GIWsuK-gdmGbPKtVdXBTBNo,2982
flax_extra/model/perceiver/_io.py,sha256=NRO2aPP9QY2FDHO3dwsZSGAgYVuR0AD_ERXmkGULVHE,9350
flax_extra/model/perceiver/_processor.py,sha256=eb_9tDwdKlF1y_Waa9O1PoZ8lds51_8sC2f1kzVmwMg,2569
flax_extra/model/perceiver/_self_attention_block.py,sha256=TfI4mg8PONKUPeaOpaFQH97Ah23dQS6_H-14_nPgvlQ,2608
flax_extra/model/rnn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
flax_extra/model/rnn/_lm.py,sha256=RrdvgR4-EtF9emloxIrX5EaepC_UjwP2dyfrQ5306ac,1376
flax_extra/operator/__init__.py,sha256=BQg7FfY-81uFQrPPF7wJcLPiBCSCrWWXVYMWGJU8-K8,270
flax_extra/operator/_rearrange.py,sha256=ru1NGnpLDkHQvTUKsQh0MkhFDE6zCfcCEKk1Y0C6MQw,868
flax_extra/operator/_reshape.py,sha256=m9L7gequtnxlQ0d7QlK0hD7vQ1ELIej066nF4QdBYCc,716
flax_extra/operator/_shift_right.py,sha256=JU_j3ZTPDO1WmS5kHDEmvaJ0Dwv-Fbw0ejwhwdmrSdA,1214
flax_extra/training/__init__.py,sha256=oGr3bwRrYYYGR9G3LlEerC0Y-Iu21uoBixd4NyZUxsg,130
flax_extra/training/_loop.py,sha256=Wol4ipqHuA5763g1V9aZlkT9ExEvEWj7hMeTJ05yoGM,19312
flax_extra-0.0.2.dist-info/LICENSE,sha256=ap9vUizhaJw4Z3pjaSHoZabb04RDq2wNYPC5wifabUk,11345
flax_extra-0.0.2.dist-info/METADATA,sha256=StQW1d5HnRVwSkOnCGekWUFO_eSeLFPl3H9BDScqtzE,3307
flax_extra-0.0.2.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
flax_extra-0.0.2.dist-info/top_level.txt,sha256=vBkctqtmseD6uWb4mnipH1ghVVsUKHXBtYh3Y0gBMb4,11
flax_extra-0.0.2.dist-info/RECORD,,
