flax/__init__.py,sha256=cUBPNBDjuMgATQy11qplLZDrQRpuBVzmU8N0OvURop0,1407
flax/errors.py,sha256=n3nyyyH8AchnV14uI9Y9BwcEiQ8TWxwYi7ZF48sUdi0,18023
flax/jax_utils.py,sha256=IAsJmQwzymFRFY-TDKO1vtyGKgU30cf0-BDrvxLH-TI,8689
flax/serialization.py,sha256=cHC-Q5DtxLhK61KR9RbdAeBH4n0FTox_gPbnE7MccIQ,12230
flax/struct.py,sha256=lIbTuAEpsM9unTaGT3eeM-ZldzfHbF9ZH8e3tYWFMCk,6062
flax/traverse_util.py,sha256=5-9E7FIcM29UHzDeF-kD9X-VWqzP1YJnMsEmPArX_x8,9803
flax/version.py,sha256=bVhPAhvWeBs_HK6fdRradU8wt3GkNWwaprb9BlgwpAg,605
flax/core/__init__.py,sha256=tOC_BFjDANBL8DqaI__upkOUsJOdtftobGMLQFfpptk,831
flax/core/axes_scan.py,sha256=0Pxr8JbeWbRzhEjkKOW5EQC3Udx_qNkmPmm4je2PpAA,5042
flax/core/frozen_dict.py,sha256=TpQJY1npoBRImUNEyTp-AZE0bLLVw9Zcxg4FJTe7qDw,5492
flax/core/lift.py,sha256=NxIdGeXFMYq-Tuo9PPLB4pO_Np5qj7NkSUAbXoJdqBo,32755
flax/core/scope.py,sha256=OW74A3VrekHf4VFriRBd769ZKLL8ymD3NiOUAdAU2a4,24042
flax/core/tracers.py,sha256=oiXGJpxmGg8H0RLB2loDZzy2H-JoVETokIkCxJZAp-U,1068
flax/core/variables.py,sha256=tuF0vaIltwKacH1_myuppXxm3p8bWrVqwOuaO6pFYEE,1558
flax/core/nn/__init__.py,sha256=2gNNB2dXTbxlJHQD75lL8Fv67CZ2mrJw_gvfrvOFfAE,1325
flax/core/nn/attention.py,sha256=XzEVtsA1S9u_vXUh_cWPZoYWYBWZzvD9foQzMefB-WQ,18495
flax/core/nn/linear.py,sha256=PnsNH9-9s_WJBMvHjUJGy1XB1GrIhqu8VIFzXYrb3Lw,12385
flax/core/nn/normalization.py,sha256=2oLU49i54o1tM9AYWlwlADNheg_6KPKNv-Gc_jx80bE,7121
flax/core/nn/stochastic.py,sha256=mqNiQ03jDPTxfQ7rjo-Em_kVz3-5N6PhW6jEEL0-WvQ,1503
flax/linen/__init__.py,sha256=VkPGFHWHN2YyMkd3Ow4rMxt__0H_vb1U9XVxzG7r-3I,1648
flax/linen/activation.py,sha256=OwMhXsk5D1jAmb6CumDpxN6TitocpBGSFePTuVa42Z0,1287
flax/linen/attention.py,sha256=yBGA9K6XrGH-Qd1CrDcD3hQh8kB3yCjXmCp8sJ8SwRg,17022
flax/linen/dotgetter.py,sha256=sGGue4jgndMuPYG1A-VYlnScfiHVhwSduxH2tYcwrVc,3106
flax/linen/initializers.py,sha256=YrbZNHdq1eh1kZzDQ4L3CsEcVME36O-LHAD-vuKrEmY,1296
flax/linen/linear.py,sha256=XC2M73XKdmbPRlCT1xpS2Kzr6b0uyN-gfJr_-UgLAKo,15540
flax/linen/module.py,sha256=R9TwaHQbtvGEioFWDSlK4NiDIfLnQXDAj1Fy8n0zqp4,49062
flax/linen/normalization.py,sha256=9xWdz9Sx81E08qHb3-SCvS3DvlnN77nzzALq1-oOetk,11184
flax/linen/pooling.py,sha256=HksJOJkptNdXvp7TJPh4au4qI7WR7Lgp1y_badJf_Sg,4087
flax/linen/recurrent.py,sha256=9x7T9s9Q7hPC6g1xH1LRDtPEeyEPxV2J9BQtCU8mJCQ,16984
flax/linen/stochastic.py,sha256=pr5lltKYC74V-fu0JSjFXU2tC9Ujf6t2q67pzDzL3Fo,2385
flax/linen/transforms.py,sha256=52k1mQIsWUalS0nUnYfKUbz8HBICpGnn2a-7zuxjTpw,21111
flax/metrics/__init__.py,sha256=fgPz9XogvfuMzYzh0WacVt46r1wLVUJO5haTG4gbXkg,582
flax/metrics/tensorboard.py,sha256=28j9vrgd4QHH60c7kACuRJPbi2TD2IzyhLhYSmNZ6sw,6797
flax/nn/__init__.py,sha256=BY7NarGK0AOl9QGbSXYJWOMY4tmGzynSWGXY-enxXNU,1670
flax/nn/activation.py,sha256=OwMhXsk5D1jAmb6CumDpxN6TitocpBGSFePTuVa42Z0,1287
flax/nn/attention.py,sha256=tmLlQDruJvGrk9CFT2E60sYxEUsCQPAz7OeyCMCk9dU,21043
flax/nn/base.py,sha256=08TZB59JJ-NG1B8FpzjoiaeMCQFtPuIf8XuMq0gRCi8,41015
flax/nn/initializers.py,sha256=YrbZNHdq1eh1kZzDQ4L3CsEcVME36O-LHAD-vuKrEmY,1296
flax/nn/linear.py,sha256=30Ko0TSHBT8Slfa5997MqiiWj8CnvmS3nDtjTQSoxKU,15469
flax/nn/normalization.py,sha256=RXVrhiNWJReqCiUf5Ha8KRIbKMudZxUI4bkQBE3OhjY,10468
flax/nn/pooling.py,sha256=HvQIGNBBsC_VlUKU0U3rI7gcKoAh3FK1kAWTx0HN8KQ,4335
flax/nn/recurrent.py,sha256=19Ks8Op69f-bp3B3ck8DmRob3eJbXBfIfzbaMto-M8U,16408
flax/nn/stochastic.py,sha256=tmZ1uSf80a-PMIXW_uxHkZcElL4jU8TaXy3sx31rkhA,4416
flax/nn/utils.py,sha256=zxQNanYvyMkEkeNBZWoggiPcvjQpS5vSvQfQ4eALAB4,3858
flax/optim/__init__.py,sha256=85IdV1aQchC5ShR6JdLniDJu-tlcnZjxWBuN_gcboJU,1453
flax/optim/adadelta.py,sha256=hk2R7A_YLdJ0ET7naDlgwOhID9U_n3R2AlC0x3skmy8,2586
flax/optim/adafactor.py,sha256=CksxmAdNnsZ74gDRMSWqp44lp_VkTQWbyrfOHenKIu8,7872
flax/optim/adagrad.py,sha256=rI8VWaDraZPSQVBJDX8v5iXG_Ij2VYxKTVz2n9CETKM,1837
flax/optim/adam.py,sha256=_AkvnU2G8VpkXrk7uSutFnMjJ_7OsA5SAUVFRgumyCo,3909
flax/optim/base.py,sha256=Yl9ahNwr1mmvh0p6d_iSZV1jizsSb2KHymCGZ-DbQk0,22583
flax/optim/dynamic_scale.py,sha256=hl9m9DI5xysyF4t4Pq5zNl8LIaJuGDfJvLhVCkCmHU0,5383
flax/optim/lamb.py,sha256=HH1khFyy6NH48W7cMqu0vFLKvoLbJ4ddXN_9Yh9jC4Q,2987
flax/optim/lars.py,sha256=G6yd0JUu2Bc3IBT5RLA3aba1mA58NOlA39zALrRppwQ,2957
flax/optim/momentum.py,sha256=yqMkOCRcTgZC8stfrB5tlOCYCm6ydzAR6AWnMHb-E1U,2231
flax/optim/rmsprop.py,sha256=dJkYJ14i0MvBKcFur_pmKroFpmKgIryPOlL0IDs-yNc,2764
flax/optim/sgd.py,sha256=h8CTAjuIHoOMSgwbcYsSR-pa4eRn8f0tmlr_7KMz2f4,1399
flax/optim/weight_norm.py,sha256=IARWofWfdi5U8GR17g07Ak_LC2FEXqQ9lYuRoiodcFU,5918
flax/testing/__init__.py,sha256=pNdmfMukh9WAVERCv2eM9iX3jACE5RDELm-Hoit8df8,1250
flax/testing/benchmark.py,sha256=s9rkLGMq6BSgFAOxx_uWcD0aShzpgaBalpiSb4meHQE,10145
flax/training/__init__.py,sha256=TSGem3ezVN8L6xotQN76irQok7DlvQ_WeSeTeqmUggE,633
flax/training/checkpoints.py,sha256=GK-4hKjFFGIjVy-_nyCnOvc1czySYPiYwcFGrRLrVxM,10949
flax/training/common_utils.py,sha256=qmafWlKR3-rU_cXdaDkhJ4KBcuDttKndH3fyrBv1lUU,1869
flax/training/early_stopping.py,sha256=sRoI1LjRFNOGG2UrzGg9lSTZLgX2fqw58Z-7Wf_1jKE,2616
flax/training/lr_schedule.py,sha256=keNEStwwIfmrAIH_7aBZmoGuTftWmtmDC2ewcOcA87c,4988
flax/training/prefetch_iterator.py,sha256=X4nrrJfWhBbl8byHiMBTVEE3npcK_e1ckWRgps0C9_4,2971
flax/training/train_state.py,sha256=s0igi6uy69raXQqzneFyi9Lt5uD25RukNYAD7TF-Fmo,3172
flax-0.3.4.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.3.4.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.3.4.dist-info/METADATA,sha256=RUTBXXrbgha5QNNMVwaDG6y2LQcec1TgcdigIgYKkBE,7808
flax-0.3.4.dist-info/WHEEL,sha256=OqRkF0eY5GHssMorFjlbTIq072vpHpF60fIQA6lS9xA,92
flax-0.3.4.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.3.4.dist-info/RECORD,,
