flax/__init__.py,sha256=Sm1pnke8OzNFhRCCXiRVL0KQODnKNgxWhM-IwzsWL9s,1389
flax/jax_utils.py,sha256=y3b4RlwLWdo_LezQLwgN7sHXEmecJNHYpEV6tBCucQg,8779
flax/serialization.py,sha256=9BwArL251TCa50cKjsh5RVkkDfRSB_HfYEGQQdZudQg,9707
flax/struct.py,sha256=BU-B-Pl11ZPfWW_p1h9yCfOVx4hebHtl7fRzm9dywE8,4656
flax/traverse_util.py,sha256=DjvP1Hoq2lwf6vFX_ZPpa0ptgpb-mdQTxCCveMo9sbg,9008
flax/version.py,sha256=iGzBUDMj_gm6I2JOotV20ksI3a2qCwrcszj24eCYZrA,605
flax/core/__init__.py,sha256=f005fYD7G-TT02q5pblOqz9FqMHM4vJdgydiwhIfBmY,782
flax/core/axes_scan.py,sha256=VkDExznja5mUaY9zeaX5o7ibFq5LR-96qHpPvivfrjg,5042
flax/core/frozen_dict.py,sha256=MGvi5i5m7caUtg2pED7OpSw6NBrXiiuI5lfslpxoF3s,3148
flax/core/lift.py,sha256=-J_Hjv5051qdEd-c1PxsGb1U15AaRemPF6cqO8-eInA,25388
flax/core/named_call.py,sha256=ChC45mHdLtLo8rC2kH-zvzketUMi8lpdKdR7USmKl5Q,1744
flax/core/scope.py,sha256=S3V7HRTIDZvJncSzaR5cedmGWCOdJ_F6ksgKGozxMjY,11330
flax/core/tracers.py,sha256=MApCbBe5YqhJkx1efcyZBAJxM2ugKrNsW2c45v_ibV0,1282
flax/core/nn/__init__.py,sha256=uH3-_QnsWY249vOg9PggXntbEUUCDLfDwPYRBfy2rB8,1319
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=0sJUqwhW_fxtyvZhS2Y-bM2hCA-RKLgbXhStuuWTJFw,1507
flax/linen/activation.py,sha256=qRmTTRm4IwIKnnoSsZ0XqUgiw-noLVCAtW_TR5cmeYc,1263
flax/linen/attention.py,sha256=dt7A-_ZfFwYI5qpVlM2u9Y_f-wqkaYKodf9M60VsnVQ,14186
flax/linen/dotgetter.py,sha256=GR7ZQDD1i7ob9UhuKidFolQBPYTNkp99WwQ1eNo2U8Y,2929
flax/linen/initializers.py,sha256=r-JvU-mTFUdwU6dX80jJweezf2Vt-Z87iEOYay5ZZ_4,1296
flax/linen/linear.py,sha256=qflwly862o0mWwYD4YvVvjlJZ0Wds_fRDIebhfbmVUQ,14293
flax/linen/module.py,sha256=9LUrIjZAl4Ob3--4jqt_OF-zugtEGmQbEPyr7mMAXmI,20729
flax/linen/normalization.py,sha256=f3J9TO6SAGcZmgPI83slELVS4c5BX_j5yTrdQ61LqZg,10315
flax/linen/pooling.py,sha256=SST2Xt0Hh2I8DJFYQtO12s-jBPpvONaNiYAAmPYswho,3743
flax/linen/recurrent.py,sha256=1c0xP2-X16g3yJbWb8wVfbCadtWsvvAcKH6L0rpSVd8,11952
flax/linen/stochastic.py,sha256=6_bVoi9QOIjYudqHxcrem8qRJq1Ton0gA0wpzUdVdOc,1725
flax/linen/transforms.py,sha256=kWt9Mqldeiir67NivSiBqoLxKiKpoo9kX18EavVh61I,9502
flax/metrics/__init__.py,sha256=wFEwierDxa4l9XWaIfHCXV28dWDubsNvHtGCxGOzjPk,582
flax/metrics/tensorboard.py,sha256=Iub-jiD9Hm2ojLveDhgkTDN2ZjRsrU25kz_aO7ciS5k,5829
flax/nn/__init__.py,sha256=VSuOgmnDg-0bh4f256XV6kojgnbozXp1gD3oH9FGS5g,1462
flax/nn/activation.py,sha256=qRmTTRm4IwIKnnoSsZ0XqUgiw-noLVCAtW_TR5cmeYc,1263
flax/nn/attention.py,sha256=dOUAlfyDmrhRox65UU41bNxezwILAcJ-U17pfxGPalk,19757
flax/nn/base.py,sha256=0jMtNNweAFr8CBY8tyAFSUpALSLKEsS5W1oTJH6y9NQ,37793
flax/nn/initializers.py,sha256=r-JvU-mTFUdwU6dX80jJweezf2Vt-Z87iEOYay5ZZ_4,1296
flax/nn/linear.py,sha256=K0m2SyzAskKpD9zzPd1Bro6uxJSVnNrHe_IrIqA-3rw,13705
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=fyMQexDs1GSyRoaJhVlahXK86UHlihLGiqcjb4hHuEM,7327
flax/optim/adagrad.py,sha256=BKWEr4YdCX8NIsxqDX1I4ccs55141GRHQJleKabty-Y,1839
flax/optim/adam.py,sha256=PWrrPqBI96wfTRPA_3bRP6weaMgeXOw3_ELEfE3a1Ss,2814
flax/optim/base.py,sha256=GVLx76NYgWdDIta5BWvwUZlMpTWctUARNE_mL12Ng2s,18245
flax/optim/dynamic_scale.py,sha256=-e4HPTa7GSOCieGzVwYwugHKSrU-XRqktMlp8Inzr1U,5405
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=Lh4GBBUITAOgkrJ4igaeeEcJhOvsBWnBkdV54ZMuJdw,2127
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=f3FTn48wGxMTnLO0eo0eW7wcUIFdb-1f_g9zBaC9Z0M,9850
flax/training/__init__.py,sha256=VwFrZBz-oYzfgyy8UNpF9oyVjq2_9_gsgr7USeKLwh0,633
flax/training/checkpoints.py,sha256=58KOndmuqkyNVW3iF4zngUAb0lfGovvr706Sv0dmUZ4,5108
flax/training/common_utils.py,sha256=IcAWzxF6zw6nmvaplqj1jRrGnS0oV78ulfejwiIQMHo,1769
flax/training/lr_schedule.py,sha256=qKvboP_wbMISHmypa1_l90WAKfEsy1pp8eW4rSJ6Xyo,4980
flax/training/prefetch_iterator.py,sha256=guhFIAWjkyDuo5EXSQcw0jCyuY6x5ATl10-Br_DGgm4,2742
flax-0.2.2.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.2.2.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.2.2.dist-info/METADATA,sha256=TeeetQjzNgqQf0Av62BDXDgnkHxyYXKxFKDGiDWI-jI,10909
flax-0.2.2.dist-info/WHEEL,sha256=EVRjI69F5qVjm_YgqcTXPnTAv3BfSUr0WVAHuSP3Xoo,92
flax-0.2.2.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.2.2.dist-info/RECORD,,
