flax/__init__.py,sha256=zk-O1o2tVERy6xGFTj5nOw2VSrTEsVOWkqDttzwAjGQ,929
flax/configurations.py,sha256=vTFFJtll1sGNKp4Q-0kZjFDUI_ZSPkTJgHGfS4NWctI,4094
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=rLmio9k9zvjgdp2ZtoxxsEXpx5c0BSMxflm0O_cbYk8,11498
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=EypUTYzp4PLS3TCVvE7ZwpZWRi_OnXomK2pYkWmuV0A,14282
flax/struct.py,sha256=6nO1n4773OkzgkREeiPhp4InwEDNj_LxleQWazeFtpo,7351
flax/traceback_util.py,sha256=71vyqPSTJfPUSbmz_INTuvYk9I8vSnachAVinXpMq1k,1989
flax/traverse_util.py,sha256=Mj67v744kK30-ZBQEfk9i18eNmYxLuZ5ShJvv1ZixKg,13437
flax/version.py,sha256=qvYUWLtZfGUl8N1R-hLVSF_hqzJMvRCcrK3oMJoT1iU,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=dbhhjUdMyTacenPMrWf2_yIwNW93Pej96fizbPaqxqE,9452
flax/core/lift.py,sha256=8jVA-l4rmXygG74-zqCxH2VwyPiT699XLxQWAy0Ge98,55971
flax/core/meta.py,sha256=oGbTHjGvq9bDjWI-6VPKYZna4SUi5j9k2aIyK-TuAdw,11590
flax/core/partial_eval.py,sha256=adJbIdE6lfJQXnqIlyOqyyEYPvr005at-4fonM5Ehdk,2540
flax/core/scope.py,sha256=EdbzAuBoxI8R_5etb3eRdDUBbiN-WX7p3US-dv6SlGI,35668
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=1u4U1B12FIp89OxP98B1XmW0SxcGGM3qPQNL4TdER4E,4180
flax/linen/activation.py,sha256=LF7yI4NZt2s6EGsF_IqCvYjpPns1JjCcAXpAdRKfglk,2558
flax/linen/attention.py,sha256=3ZJ8Hq2NfO57KxDxhD9eVUnTNxOENe7qe2y6-AMpdxA,18548
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=kRSnrryAn9IYNTFXmcqt1jxGMo3Un6mllSqeoLlgfxs,7553
flax/linen/linear.py,sha256=9SbiKGDXYzdDIdTjEESOmj-nFuqhqWC-nOWmkgi1Krw,33001
flax/linen/module.py,sha256=v7B74b25SfPdHD_smhaHrNOo0CNjFa86mlhswgwzhQA,88743
flax/linen/normalization.py,sha256=IN9Fpu3J3Rbft0wl-tBCHTvX9Ofqo25UVkBbPrBoOJo,21679
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=K3Elfei0WZfjm28Uj0F1A-sXb4ZcFfcoFg0xD4kCSHo,11459
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=RoQkp8nzyyEMPe-IvCFjhVe4jXd0c2bpy-YFIkKa2DE,11490
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.1.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.7.1.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.7.1.dist-info/METADATA,sha256=84Npw88za_ioHh9w4J-DtKCjwEJmZb_mLhnJ9PLPnlY,9980
flax-0.7.1.dist-info/WHEEL,sha256=AtBG6SXL3KF_v0NxLf0ehyVOh0cold-JbJYXNGorC6Q,92
flax-0.7.1.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.7.1.dist-info/RECORD,,
