flax/__init__.py,sha256=cUBPNBDjuMgATQy11qplLZDrQRpuBVzmU8N0OvURop0,1407
flax/config.py,sha256=eu6i8nMSy76_4Tk358Cz5NFgLIysxWhHj4fKirMMs9E,2005
flax/errors.py,sha256=lUyZK2MqzPasITfb0jjpJz6VErUtIse5UdwY9kSPLJA,19225
flax/jax_utils.py,sha256=PlwQPFVJQLMFFMWX2uA0orzwqD8WMmFb3O1RzTuV6bE,8107
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=Dwjv80w9h_kK9ECowG4iOq-Tnq0q6KIKLJbZcwA0Um4,12733
flax/struct.py,sha256=jXMufcRkkddwpHm1E5bhXwXYCk8NNy24iXPbVu232Zg,6416
flax/traceback_util.py,sha256=7w8vGRKn36r7Ty0rpzcLj90aqeDHwxi1GfmIfXpzWGQ,1989
flax/traverse_util.py,sha256=Uswp2HULTSJYRS9Ks1XtHAoRL_9SJHt8v7vwrz_xbB8,11596
flax/version.py,sha256=m-Eb4gUndHHEO2_ozaiRSAuLDIrzzh8odVPNWNL0EHI,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=b8AC-kZAhAk1dfKTmf0Dp6u5AU8MCL8_dSQQBpaQXuI,6005
flax/core/lift.py,sha256=lwvc4xGMpajtifPr8cdfxOkSGIJ2TYjmum4L6AsqR6c,34800
flax/core/scope.py,sha256=ja32DKRZ3IZ9S0A-RSmZOj3353wOcX9-WOANNlz714A,24150
flax/core/tracers.py,sha256=-o-3Lez6dxHpUQBxtRuP1XWLBKdNwIx1glIHyZPQFBg,1054
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=cPJizCNFqHgmO3Bc9CuDGDTEkHVYcxZvV0DRlY65Ozg,12502
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=nwf--Zckf_TAjb2Dlv4tvMVl5GN9We6ZZZY3sVwdpT8,1734
flax/linen/activation.py,sha256=OwMhXsk5D1jAmb6CumDpxN6TitocpBGSFePTuVa42Z0,1287
flax/linen/attention.py,sha256=lvQO_zvVE2LLmOle2iDUz4Wzoy015c80_AOhBGXqxSw,17587
flax/linen/dotgetter.py,sha256=sGGue4jgndMuPYG1A-VYlnScfiHVhwSduxH2tYcwrVc,3106
flax/linen/initializers.py,sha256=g21xE7PnpiKHnUWuf3ctlmmuNWxT8I8cDQkSIyIX5NM,1474
flax/linen/linear.py,sha256=I8h9x8Oue6cSEQu5ODFMl2_MdBxPJ2jGER7QZjtQejg,16187
flax/linen/module.py,sha256=PE1pJc5iB9ILIxhm2hezIOsYwvMrwwcWHWcDcNTaWTg,53621
flax/linen/normalization.py,sha256=99Nx3Fw7G2cjKUb2pO2sLkdq-dIvdkcIJ4NwEh3Zei0,12455
flax/linen/pooling.py,sha256=KWZPE-jfl5gL4bcDFhvncIvgjM0awpYM8HPT0YdKZE0,4926
flax/linen/recurrent.py,sha256=uCo0F_sz_nTW1KJcgE3WW7LB5Kg4tVpAKhjJQNrPfeY,17023
flax/linen/stochastic.py,sha256=pr5lltKYC74V-fu0JSjFXU2tC9Ujf6t2q67pzDzL3Fo,2385
flax/linen/transforms.py,sha256=4vFKwozKaywTOK8Xy37B5dDN7UiLkoxyg-pE3Owcl9Y,26980
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=AZBYKnlY7p4FrOcPMh8uNnlhV8Q4S0nv5h-DY4Imo5E,21033
flax/nn/base.py,sha256=8KRWtHGfaa7rEqwizwxafls_Vkxnr1yR7dXRPW3QziA,40977
flax/nn/initializers.py,sha256=YrbZNHdq1eh1kZzDQ4L3CsEcVME36O-LHAD-vuKrEmY,1296
flax/nn/linear.py,sha256=6mcuxlncMa9hbJH5YmiC3oyWUi341EQEkFjvTAwbBWY,15570
flax/nn/normalization.py,sha256=Q9tIGlXt5jDC9gjRYIcmK1lCWKaCKl1kyMIrN9_P2Hg,10485
flax/nn/pooling.py,sha256=L3WJ6nqnYw9k6YGV1kh4JmDeTw8UYyXTLoBzmWHc6xA,4329
flax/nn/recurrent.py,sha256=AxFuSpLitSVtMNyT9uHwUZzCmb8ZUACm_9-sFiHct2c,16435
flax/nn/stochastic.py,sha256=xu2GvrleYhKW5emCUhRO5xzGSu17oDs-Lkno96q7UYM,4406
flax/nn/utils.py,sha256=KjxP6YhaQJsoCrWJdPqKGqKzaBhfX5Y39Qdcxf-X9r0,3832
flax/optim/__init__.py,sha256=Um-Dwf3C9b_glnH1kfrzc2yLLawtDlSU8hOE5kVHdMA,1902
flax/optim/adabelief.py,sha256=oQj0sKPX7PazmNRlYVAam5r7qqLgoQf6uHE-0yRwgQ0,4420
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=iypTeoSpc7VGe0q3xf4h8YW65_uNBVsQGQ1VieHMUfo,20350
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=x9In7xbIQ9F9pZHanez1-bS0AikOPOHNARC8pRyGvfw,12710
flax/training/common_utils.py,sha256=rFwjHzf94PQCLZBC_rh6JBbzZvR1VEf2YLJdBiHxj-E,1899
flax/training/early_stopping.py,sha256=sRoI1LjRFNOGG2UrzGg9lSTZLgX2fqw58Z-7Wf_1jKE,2616
flax/training/lr_schedule.py,sha256=pTlAObA2mYmxtZQH6UcPeKfqz4hwK6d4Oi0c93eXR3k,5414
flax/training/prefetch_iterator.py,sha256=X4nrrJfWhBbl8byHiMBTVEE3npcK_e1ckWRgps0C9_4,2971
flax/training/train_state.py,sha256=J4IFdeWQ9UbCFMFVN6LV6GTU1sITSEhrful84hWpzxs,3256
flax-0.3.5.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.3.5.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.3.5.dist-info/METADATA,sha256=T8f8MBXe40XXTzAwVZoeWidcs6iCeeaEfcc9yfJmwsk,8665
flax-0.3.5.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
flax-0.3.5.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.3.5.dist-info/RECORD,,
