flax/__init__.py,sha256=Sm1pnke8OzNFhRCCXiRVL0KQODnKNgxWhM-IwzsWL9s,1389
flax/jax_utils.py,sha256=ZYA5-vKgl4RTzWvw7W_3wYtxeRmiENmU9tiLJ4XicGc,8778
flax/serialization.py,sha256=d-VbyjVICgUZiIVVmJFMwDGtZ5Td3acQrX9NZVK9YnY,10170
flax/struct.py,sha256=32mHCKnSzSoqvl5o3qJM4Act8g7QN8-KoprJCqe8Gmo,5263
flax/traverse_util.py,sha256=szyb57RcC8QfcbZSOnmXdU5Ki0GuinAeNhgJILUEeHg,9545
flax/version.py,sha256=yvr-8Bp_LrvmwpgztuKSyTcCWdCrz2cngJNnXDo3aOs,605
flax/core/__init__.py,sha256=f005fYD7G-TT02q5pblOqz9FqMHM4vJdgydiwhIfBmY,782
flax/core/axes_scan.py,sha256=VkDExznja5mUaY9zeaX5o7ibFq5LR-96qHpPvivfrjg,5042
flax/core/frozen_dict.py,sha256=rmja1wR-zJumuGlv9fNgzzHcGvGHZebjfKHMfR0Emnw,5162
flax/core/lift.py,sha256=DF_9ccE7sdmsrinuXSLmtXoJEEZUx5H8ycl5SfqZ65o,25819
flax/core/scope.py,sha256=-h-0Bo8Jow5wLQRpjBOJH4JDGHRt4vWVB3GAQjggmjg,21851
flax/core/tracers.py,sha256=4Wtb-1iwCFfaEDM9FF2nLrFSgM0l32MuxPP4TLYfFSE,1068
flax/core/variables.py,sha256=PfQ4zX26GKZVT6zlMfUwOqB3phdizFMRVyGZuooiNI4,1558
flax/core/nn/__init__.py,sha256=V_WPakQR3FipQtMuhwyapktisWDUsphWmHaF-mQh3aE,1325
flax/core/nn/attention.py,sha256=Fvsvbpc5zYBfn72Lurr87HbvKrQhiPFio3kExPRpR7k,18506
flax/core/nn/linear.py,sha256=ryzOxNNUIkU-610QIRsibE6b16XUzECpVOEJUL-lRRo,12395
flax/core/nn/normalization.py,sha256=RkotjtjLAi0_fwj3NGhHUFzx8OlitrFIdbq_PaKSuxw,7121
flax/core/nn/stochastic.py,sha256=LDdr3-3qYfQdwbNn6RSc2kVoNCOvuahWzMRmc1SeHjE,1503
flax/linen/__init__.py,sha256=PSh5aoLBMh06vYBI7F4L79wHvA2rA_nr7Fdbpg8JrFM,1559
flax/linen/activation.py,sha256=NDdIl_Lq-57wRRWznfmKqzoQg256DHOL79iO_MN_H6c,1287
flax/linen/attention.py,sha256=pk0I17ij-H2b0YHVi94f6QGmXMrSMHgIj6IdTo5Xb9c,14217
flax/linen/dotgetter.py,sha256=YFGXMEPMCyXzDKf7ru4o9jV9nmg80mOAunK_qrzw2l8,3068
flax/linen/initializers.py,sha256=r-JvU-mTFUdwU6dX80jJweezf2Vt-Z87iEOYay5ZZ_4,1296
flax/linen/linear.py,sha256=enQCA5ekYlRq6_gaiE-OY6RO78g2A8P0DyIYmQpPhmo,15420
flax/linen/module.py,sha256=FUzoH9aZ42mD20xCqjlfiDfjJTz-4UrP92QaNNsyLt4,29802
flax/linen/normalization.py,sha256=9wf2q6vk4PYhajNAkEPMT0PA0BJuwDUwcjHGu7D6F4k,10322
flax/linen/pooling.py,sha256=SST2Xt0Hh2I8DJFYQtO12s-jBPpvONaNiYAAmPYswho,3743
flax/linen/recurrent.py,sha256=769LlffGu5ntAVZUP5V7YpszS3-1Rc1lzevnRwDYnrA,11817
flax/linen/stochastic.py,sha256=xhnZKJrlEZ-fbcuchVRkHZS_jw4xA1tVy6mRXssXOAM,1911
flax/linen/transforms.py,sha256=orCHET49it1guGWjHU4Cu9LA3gvqPrgsZ0TBRfHSTvQ,10152
flax/metrics/__init__.py,sha256=wFEwierDxa4l9XWaIfHCXV28dWDubsNvHtGCxGOzjPk,582
flax/metrics/tensorboard.py,sha256=Iub-jiD9Hm2ojLveDhgkTDN2ZjRsrU25kz_aO7ciS5k,5829
flax/nn/__init__.py,sha256=6VHh112dAAlMRa5b5L7cQCQ1Kkz_AzBCQRtkcgjrTZ0,1468
flax/nn/activation.py,sha256=NDdIl_Lq-57wRRWznfmKqzoQg256DHOL79iO_MN_H6c,1287
flax/nn/attention.py,sha256=viR-_KLTn_ABDuk_tb0QU-w8M83iJPoihkey-2mqJ_w,20158
flax/nn/base.py,sha256=iYuV4NS1i1mDTTOg2boKVDYEcxHcY0Li2bihvfE28qY,37972
flax/nn/initializers.py,sha256=r-JvU-mTFUdwU6dX80jJweezf2Vt-Z87iEOYay5ZZ_4,1296
flax/nn/linear.py,sha256=SZIR3zqKWfs2AZC_-ydrTMzromft_lUrBneuytyt5Q8,14092
flax/nn/normalization.py,sha256=qDevLgKSVGXmNpgdm7Am0Kd4ZzkDtR8gjmMnGgMFRzg,9874
flax/nn/pooling.py,sha256=SST2Xt0Hh2I8DJFYQtO12s-jBPpvONaNiYAAmPYswho,3743
flax/nn/recurrent.py,sha256=rHaK_QYgS8Fez5trktodgmpfVcsuoYqsHSR0tZcaHZQ,10703
flax/nn/stochastic.py,sha256=U8qbQK54eNtGEZxUWX3H2tn64BSc_aLBZRwiNO0shHc,3426
flax/nn/utils.py,sha256=IriyjMsPmnw9jwJl63ep_snLOwrqxdMNJc-Lml4f7RM,2473
flax/optim/__init__.py,sha256=hL0oSxp5ZG9czMvpXopgZolKy_sfYuHAiq5CsmhwbMo,1406
flax/optim/adafactor.py,sha256=Gg_0d8Kcp5yAgH7A4HH__PkZ24j-mxzTsv7BMjgvI00,7643
flax/optim/adagrad.py,sha256=BKWEr4YdCX8NIsxqDX1I4ccs55141GRHQJleKabty-Y,1839
flax/optim/adam.py,sha256=PWrrPqBI96wfTRPA_3bRP6weaMgeXOw3_ELEfE3a1Ss,2814
flax/optim/base.py,sha256=r9NXfu2YOvUndsAwsfzsXjr40zW5sODrO9AAlSZVoHw,18522
flax/optim/dynamic_scale.py,sha256=ghyzs_Y7OVjL43WxKRTWcEJ6S9Q78zump163aK2XDZI,5402
flax/optim/lamb.py,sha256=5jbbzvH3KO-Tl4kP0qjV6KfBpI__zZZ6PN3jyJfuPa0,2961
flax/optim/lars.py,sha256=_mLIJXLy4OCfTAiZo7siDDIfrmsZOWK0cUwOLX54TTM,2964
flax/optim/momentum.py,sha256=XDa0BrHjRHN4kCXI1ilqfvk5die1oEUVB41Oi5U8gsA,2236
flax/optim/rmsprop.py,sha256=PmM6fFAde0-hxofdo6TKvfEzYCvwjZcbFQY3dMhroZI,2767
flax/optim/sgd.py,sha256=DV1OEhVrHyZG3VqEr3nFNAp9Hge4BNrsYNDrloCIZKE,1401
flax/optim/weight_norm.py,sha256=XqcqFpm7ZRFNpA3Z_FaMAD5BnVQmdhD_fHGSw74vlhc,5913
flax/testing/__init__.py,sha256=ZRv8pGRJ52783jPvKkujUsyX5YAZ_pc8zgTgLt_0XCM,1250
flax/testing/benchmark.py,sha256=u0g1ED8yqRKtZCOOtptNnv44KUzIkAOI1imwOlU-k1E,10145
flax/training/__init__.py,sha256=VwFrZBz-oYzfgyy8UNpF9oyVjq2_9_gsgr7USeKLwh0,633
flax/training/checkpoints.py,sha256=Xs9u9jD4zYqStviky-Hg5KwC-cgwk2UGcP41YqTq2j8,6457
flax/training/common_utils.py,sha256=IcAWzxF6zw6nmvaplqj1jRrGnS0oV78ulfejwiIQMHo,1769
flax/training/lr_schedule.py,sha256=qKvboP_wbMISHmypa1_l90WAKfEsy1pp8eW4rSJ6Xyo,4980
flax/training/prefetch_iterator.py,sha256=hsLPdWScuhDmhWTnvgVUPXMuEb4YxjkAlfOnaYTT70s,2967
flax-0.3.0.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.3.0.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.3.0.dist-info/METADATA,sha256=2iSeNlwq8WjKd6NlJ8ab2jpaOCw7UjxjfWe9P73MeIM,6976
flax-0.3.0.dist-info/WHEEL,sha256=S6zePDbUAjzMmpYOg2cHDxuYFWw7WiOXt6ogM6hIB5Q,92
flax-0.3.0.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.3.0.dist-info/RECORD,,
