flax/__init__.py,sha256=zk-O1o2tVERy6xGFTj5nOw2VSrTEsVOWkqDttzwAjGQ,929
flax/configurations.py,sha256=vTFFJtll1sGNKp4Q-0kZjFDUI_ZSPkTJgHGfS4NWctI,4094
flax/cursor.py,sha256=UGxwteRJG1zaG-bbw7yOIoihUecsGidsicdMOYlRvrg,13735
flax/errors.py,sha256=YZxFp6jryi8KfvDhhPkIAKGpJw3cD3G_pMSFjF7MdT0,29167
flax/ids.py,sha256=qQrbtxu1ZYQXJNGijfLWXvAZiOqCEtdTBXsuwFJJ3GI,1752
flax/io.py,sha256=pcoyzMeeQgnGNWecYZN5L_W3Y31BZCTDyDThdoxpUQM,5317
flax/jax_utils.py,sha256=pjw1Pi8dYoS8y5r5TWri1sGiuo0rShsLc4oZ22tP1wE,11519
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=qvbitRNplXE_HkC-P5o0HeVef580arq3mbpIBW0u3oU,14244
flax/struct.py,sha256=ItOoqEz4HIo0VnD8U3vVaOGEesZVB-ReIBRvPPkl7fM,7349
flax/traceback_util.py,sha256=71vyqPSTJfPUSbmz_INTuvYk9I8vSnachAVinXpMq1k,1989
flax/traverse_util.py,sha256=Mj67v744kK30-ZBQEfk9i18eNmYxLuZ5ShJvv1ZixKg,13437
flax/version.py,sha256=O567QVY1Hr4SRLO3SOyolQdKIR_78WQfMsu0IcielHI,650
flax/core/__init__.py,sha256=CgxSNB7TSUnF8ChJ1srQcQagxNqiRDw94Of1t2ESSEM,1439
flax/core/axes_scan.py,sha256=FTY0kRKxewl7mFmIbcnbK-Dm2kDUhJshZnDyQ_mjRTs,5406
flax/core/flax_functional_engine.ipynb,sha256=Fm9guf0xme8voulyRpqhCyWWyir4IUNcZOetCDTBZjo,10412
flax/core/frozen_dict.py,sha256=1VOFpbM-3Vm5tegYBIsGHD3zPE090JRgPSsoqGSzY9s,9571
flax/core/lift.py,sha256=nrr52Ixm-JsdXYjhgG6jGe7DDJc8qXnobP42uNGElcc,56084
flax/core/meta.py,sha256=t506uMPqCKd8YEqXv3k2I2iojP4DOZ9zTIylnH7gj64,11677
flax/core/partial_eval.py,sha256=adJbIdE6lfJQXnqIlyOqyyEYPvr005at-4fonM5Ehdk,2540
flax/core/scope.py,sha256=IbVDZ9R5olHtOhon7bblG3ghTdl1gVL_VdVNYlkxdZI,37717
flax/core/tracers.py,sha256=Oi656iLKyEK0U1xXNhLFCFTnO2x6hn-FUmOqoot-sBw,1053
flax/core/variables.py,sha256=ki6bPJsAtuUNSuQGMgGcv67a4pq8cALK4C08QDQ48Ao,1558
flax/core/nn/__init__.py,sha256=-kZvNSjKI9UodCqxoTUKNHh2hXh9Fa3OcVIRltSB81Y,1795
flax/core/nn/attention.py,sha256=cZFbbpWcBBUHX6PsClJ2GrObV5Dl8cdeBkyu7g6aiAQ,18314
flax/core/nn/linear.py,sha256=IK9hRDj4rAsXwaInPUcYaK00r01Q6dP1sK0HoubPz8Q,12204
flax/core/nn/normalization.py,sha256=drOaYDY9FUPwA96ibIqtaM8J9g3OFFKw_BG10gMDTYM,6974
flax/core/nn/stochastic.py,sha256=6DXm-imh4w5gllk5GVlFFmOToYDfeVDMTPtNgZZax14,1503
flax/linen/README.md,sha256=lOmtF3n90VwTrpZDremYhxpjW2eS23mids9CLYC2FOk,2201
flax/linen/__init__.py,sha256=cdcfIRQUun9Xtj0NFbbs-g6N7QEL1LbdKTdYoKKFcOk,4224
flax/linen/activation.py,sha256=LF7yI4NZt2s6EGsF_IqCvYjpPns1JjCcAXpAdRKfglk,2558
flax/linen/attention.py,sha256=HQ4-Z8KtDYGwMXSL8pbkUlAl2vGwU3MaSNAzZAtXE10,19268
flax/linen/combinators.py,sha256=6KIPzptVATDfx0z8yUR2TI2TpGvmwGqHLm4fPfVAKus,3225
flax/linen/dtypes.py,sha256=IYBgJRgaEtMGa_njdN-VUrMTUuavYknFUA81O7UpPuE,3872
flax/linen/initializers.py,sha256=D8cNhLQ-Tfl-FuMUU6MQXV1Mm7NrnbXiq0mOjqBU6Gw,2622
flax/linen/kw_only_dataclasses.py,sha256=87U_HA2W2_aSoKu5u8AoaucMQzQb8zCX20ia52a-5BE,8868
flax/linen/linear.py,sha256=a87J6LCUQVKdNZixIckCexckSO9nbsNK6XKmxUmtY30,33810
flax/linen/module.py,sha256=e6ZrKB_hfs0bJlH3c1_nGzG-fLrWoJPh0Sh_OzorHyY,95157
flax/linen/normalization.py,sha256=XIGRbb51OYtfWlwzqdeIF8BRnz1ViTnObleTl0iK1nk,22488
flax/linen/partitioning.py,sha256=KYckTwEimZsKHFdLKBCE3tc42-4D0ANDd9PfoaZZY28,19344
flax/linen/pooling.py,sha256=UjPj4AAoGnMWX4cfgfL8FQ7YAuGUXmWb6AD9Mw1JVU0,5509
flax/linen/recurrent.py,sha256=ZPZb1N86LGnc4KAY7pR7jlrQowbJUHTDjIu0bwt-PHY,38023
flax/linen/spmd.py,sha256=TcOJ2u3MDDoT6PXXkiB93VTUfDrdkpajbkFbCjob0ZA,11508
flax/linen/stochastic.py,sha256=St661-57lno-ERtPcgX78TboV74aZzRl4VX9wgBac9Q,3041
flax/linen/summary.py,sha256=nqWZ6JPd9asqlagswnnOcwFxMgBghrQplFnY5NuAJVY,20309
flax/linen/transforms.py,sha256=nCPC043JHfIhTSI0b0sz4i9N1yHBRBfER63N3kJ-His,59542
flax/linen/experimental/layers_with_named_axes.py,sha256=5bRZ8D_V4rcvaL8BGLdp2tnzjpKfE2eDDTP6hToGggA,11680
flax/metrics/__init__.py,sha256=qEgClvMWm4hzoAPfpJiej6gXqpBZ8b8bPZrLB1BRUU0,1163
flax/metrics/tensorboard.py,sha256=HrVu4dUay4XmFDbjFvuwk0IKIqC6Ta11G1e9RHRJthY,7804
flax/oss/ .git-blame-ignore-revs,sha256=eJmLQj4P4rVEN8uRkflxwOhLFte29Y0WeGoI1uRNWfY,443
flax/testing/__init__.py,sha256=_OTLJJ6QRBv5CH6a3oK8MpUjEwd_OnMyzyvC6HleIgM,647
flax/testing/benchmark.py,sha256=YDMcZ10im3Q9wcoX7gDF79MS0X2EPeMtLdvovw0PXAU,9452
flax/training/__init__.py,sha256=wXT2B9npbmhWzD-dGXNkWQwDU-JpoUxOni-qiZVP0KI,613
flax/training/checkpoints.py,sha256=3hPqzRFQhdYcOhrEodl0ckqwPHgYIKvycUpQq3PXcok,43203
flax/training/common_utils.py,sha256=0CFVdxcb8TzhbRE_oa8_rQqOw6hEPS8T0xLhkKkxiWE,3692
flax/training/dynamic_scale.py,sha256=OsILVOJTU65la0757SJ1w9JFnh3sbOsUOugJ5CVlb4c,5835
flax/training/early_stopping.py,sha256=t58z8sgtsBttCazDLOZmj6IIHDq6wXWDkL3avFrBjz0,2662
flax/training/lr_schedule.py,sha256=rU1e-QFBMAKuXS0HSAMokoavHym2TLoi8y_oDVGgCFQ,7429
flax/training/orbax_utils.py,sha256=JTzkeqNj099C7i75ORrhwxn3SXVh_iGKPCpW7IO2VeI,3714
flax/training/prefetch_iterator.py,sha256=3Pi-n1p_ioCT3t_ZZDLhQWwfg49BMy4MAtzYuUCEqX0,2984
flax/training/train_state.py,sha256=oHaqpZ9WLdpm7xFW9yX_rFDTrDw3v0HH06GGfAptAiU,3314
flax-0.7.3.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.7.3.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.7.3.dist-info/METADATA,sha256=ZPxgrNENKSSNZjDsYvmAC44SGJxVQTCN7pN8535cXLQ,9958
flax-0.7.3.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
flax-0.7.3.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.7.3.dist-info/RECORD,,
