flax/__init__.py,sha256=3cz1v6NVRNbr9QhGisphpGCqTItfXBD9qKPvl8TDDPM,1417
flax/config.py,sha256=12s5ZOn8D84ReybGJdopEXTvty_yuxrlCDCID_KUvMI,2009
flax/errors.py,sha256=i6bzw5aECRBogj76WWYSZS6OGQQSFtpJRELqkOUOX-M,19233
flax/jax_utils.py,sha256=dDDNha0Lf72LUINOjl8cXA-bty1TrWLSbgTcESvPs34,7660
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=Dwjv80w9h_kK9ECowG4iOq-Tnq0q6KIKLJbZcwA0Um4,12733
flax/struct.py,sha256=VGxgsigvEs5G_bbiB5Dzw2TyCss2tjoCjd-SURyeBkQ,7376
flax/traceback_util.py,sha256=7w8vGRKn36r7Ty0rpzcLj90aqeDHwxi1GfmIfXpzWGQ,1989
flax/traverse_util.py,sha256=Uswp2HULTSJYRS9Ks1XtHAoRL_9SJHt8v7vwrz_xbB8,11596
flax/version.py,sha256=ZpIMK8ayDEJV1pz3n9lyZoGsj8c6XezssQZ7or-V60c,605
flax/core/__init__.py,sha256=tOC_BFjDANBL8DqaI__upkOUsJOdtftobGMLQFfpptk,831
flax/core/axes_scan.py,sha256=WbEg8GVaytAa3uixmtzkYPBh7M5VL5zIswanlKpqCok,5045
flax/core/frozen_dict.py,sha256=b8AC-kZAhAk1dfKTmf0Dp6u5AU8MCL8_dSQQBpaQXuI,6005
flax/core/lift.py,sha256=Q-Od4fm5YQL9WwelAfKE4F4ORHQyz9R2etUVptfCMf8,42893
flax/core/scope.py,sha256=6Ff_aF5zkusUHG2MjN29D2TnQdWkqSiRaay-w3iEezg,25083
flax/core/tracers.py,sha256=-o-3Lez6dxHpUQBxtRuP1XWLBKdNwIx1glIHyZPQFBg,1054
flax/core/variables.py,sha256=tuF0vaIltwKacH1_myuppXxm3p8bWrVqwOuaO6pFYEE,1558
flax/core/nn/__init__.py,sha256=5pHpycQ6q2eOJAd-gv0kJWEtiatvloi8iRxEFK00Dnk,1434
flax/core/nn/attention.py,sha256=QkOzuE85DVSNBAYXKqYhQ1dmqXEjr9d9IsKcWhwNmbc,18503
flax/core/nn/linear.py,sha256=iRS8UFqVYJ1UX2EsSa7IPJO_z4PSkEeH8KUlXReQVqU,12509
flax/core/nn/normalization.py,sha256=ErrEDyOU4hpUB8N9lqGm70D16hTtzEAua6JTgkthO-c,7132
flax/core/nn/stochastic.py,sha256=mqNiQ03jDPTxfQ7rjo-Em_kVz3-5N6PhW6jEEL0-WvQ,1503
flax/deprecated/__init__.py,sha256=fgPz9XogvfuMzYzh0WacVt46r1wLVUJO5haTG4gbXkg,582
flax/deprecated/nn/__init__.py,sha256=BY7NarGK0AOl9QGbSXYJWOMY4tmGzynSWGXY-enxXNU,1670
flax/deprecated/nn/activation.py,sha256=OwMhXsk5D1jAmb6CumDpxN6TitocpBGSFePTuVa42Z0,1287
flax/deprecated/nn/attention.py,sha256=_tEbitnFP8fsMu7MUWOBu2XzLQApnYc-Kis2ti7OkKM,21087
flax/deprecated/nn/base.py,sha256=8KRWtHGfaa7rEqwizwxafls_Vkxnr1yR7dXRPW3QziA,40977
flax/deprecated/nn/initializers.py,sha256=YrbZNHdq1eh1kZzDQ4L3CsEcVME36O-LHAD-vuKrEmY,1296
flax/deprecated/nn/linear.py,sha256=6mcuxlncMa9hbJH5YmiC3oyWUi341EQEkFjvTAwbBWY,15570
flax/deprecated/nn/normalization.py,sha256=Q9tIGlXt5jDC9gjRYIcmK1lCWKaCKl1kyMIrN9_P2Hg,10485
flax/deprecated/nn/pooling.py,sha256=L3WJ6nqnYw9k6YGV1kh4JmDeTw8UYyXTLoBzmWHc6xA,4329
flax/deprecated/nn/recurrent.py,sha256=AxFuSpLitSVtMNyT9uHwUZzCmb8ZUACm_9-sFiHct2c,16435
flax/deprecated/nn/stochastic.py,sha256=xu2GvrleYhKW5emCUhRO5xzGSu17oDs-Lkno96q7UYM,4406
flax/deprecated/nn/utils.py,sha256=KjxP6YhaQJsoCrWJdPqKGqKzaBhfX5Y39Qdcxf-X9r0,3832
flax/linen/__init__.py,sha256=9W_-PvE8fLk0UbthJsjmlOzzK9KoSN-yHvp4pmgMkTQ,1795
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=hHB-nevzPtiSygHgR9l6dKhCA9vAIBlNE7-_vI_VRaQ,53697
flax/linen/normalization.py,sha256=os3jLmM0-OzwO66ISUUsya4dMcs1QWIQuMTVQZG8Bpw,13961
flax/linen/partitioning.py,sha256=IyG7sg4molPkICzdVravZD2G-6BzGr-wunS0ZSxNCtU,20317
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=MdauMdtbgk6e4k44q0p-ZAm5w094rYyt3USGDqyCFXc,38915
flax/metrics/__init__.py,sha256=fgPz9XogvfuMzYzh0WacVt46r1wLVUJO5haTG4gbXkg,582
flax/metrics/tensorboard.py,sha256=28j9vrgd4QHH60c7kACuRJPbi2TD2IzyhLhYSmNZ6sw,6797
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=3LJ0TfHjMJW5w07M7Nmq8BnIn8RuFEGCTGzrvyszXQI,20327
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.6.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.3.6.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.3.6.dist-info/METADATA,sha256=KDseJ6AJ7nFv01tUNTgcsfWru730OLXSP-4cK8ic7Vo,9294
flax-0.3.6.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
flax-0.3.6.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.3.6.dist-info/RECORD,,
