flax/__init__.py,sha256=zk-O1o2tVERy6xGFTj5nOw2VSrTEsVOWkqDttzwAjGQ,929
flax/configurations.py,sha256=vTFFJtll1sGNKp4Q-0kZjFDUI_ZSPkTJgHGfS4NWctI,4094
flax/cursor.py,sha256=YTLT1sHd32iA3FAAF4rDIp_KmtjA9YYeLyaYiEO62Fk,25689
flax/errors.py,sha256=D2_IKvlotd0jk8Kad9lULXLsrpju7SJIHFX5RRSKopE,30804
flax/ids.py,sha256=qQrbtxu1ZYQXJNGijfLWXvAZiOqCEtdTBXsuwFJJ3GI,1752
flax/io.py,sha256=pcoyzMeeQgnGNWecYZN5L_W3Y31BZCTDyDThdoxpUQM,5317
flax/jax_utils.py,sha256=idzEUDyxNfplHKQXQWnkuZfE6ARMCP9UjdFiJS7g9Gc,11526
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=qvbitRNplXE_HkC-P5o0HeVef580arq3mbpIBW0u3oU,14244
flax/struct.py,sha256=Z7MBeUgaWcQetTXreSBcJlr61_Wj0ZYFFV5R8GewUGY,7320
flax/traceback_util.py,sha256=71vyqPSTJfPUSbmz_INTuvYk9I8vSnachAVinXpMq1k,1989
flax/traverse_util.py,sha256=_abakK4yW1ESAK0yR8eYLSZXtFPrMi3xoicWx_cxjms,13431
flax/version.py,sha256=V8PuItwQG9WOgLljvWYQG2OxhMDt2E7N45aIRbcbTS0,650
flax/core/__init__.py,sha256=CgxSNB7TSUnF8ChJ1srQcQagxNqiRDw94Of1t2ESSEM,1439
flax/core/axes_scan.py,sha256=EJmJH5wtlzyar3tPDx5Sx2Ia7TZBNjM9mbGcAvunJro,5413
flax/core/frozen_dict.py,sha256=_wbXQeLqucc5fPRvPJjr3kzPE2w3jlQh-mlE1SMdFAU,9561
flax/core/lift.py,sha256=17zDRVmUsUgkngx9DUxDK9_K2DyCZqqLuccJJvtVzz0,56364
flax/core/meta.py,sha256=2OYNx6ncErYrMUOYUYX_HTEq3menI4fo-tOiAbV1B2Q,11659
flax/core/partial_eval.py,sha256=lEpztHKkXRE7DmherCNkFBCL-yoQ_aDxO0K7zAFN5fc,2547
flax/core/scope.py,sha256=EGRoSI36iu7eP4gCSO5xGDKj00Wmgl7BbSe50Kr1OjY,37657
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=AbbvF4-FEIplRQRimBcr9zvC2ixUT-Uh6r160-_Omv8,18246
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/__init__.py,sha256=OM6dCaFFy-I_wrcyAUXdwNSVPAOS3lW8C18SL67fapc,4367
flax/linen/activation.py,sha256=LF7yI4NZt2s6EGsF_IqCvYjpPns1JjCcAXpAdRKfglk,2558
flax/linen/attention.py,sha256=KBDTF3LUqqJ81GuKxph-4GuOl8bflhGcD-hkveiIB_E,23063
flax/linen/combinators.py,sha256=6KIPzptVATDfx0z8yUR2TI2TpGvmwGqHLm4fPfVAKus,3225
flax/linen/dtypes.py,sha256=1IiEPSzNj5GTndkY85G6mKp272FmVyB5GeJ6ImssyVM,3861
flax/linen/fp8_ops.py,sha256=OSarjZO5Q23_zBLFSXQbkz-PLvCyYPeGH-QALJVCLFQ,5865
flax/linen/initializers.py,sha256=i1RGTrd6OFwsBPUDSHBtBvqbZJJZssYBV_j5XKiFCd4,2683
flax/linen/kw_only_dataclasses.py,sha256=87U_HA2W2_aSoKu5u8AoaucMQzQb8zCX20ia52a-5BE,8868
flax/linen/linear.py,sha256=JVWm6RjJ9PqpkgTg-6qslV4JcyjddDwEk3aztEMoECM,33738
flax/linen/module.py,sha256=mJEWwlbc-U83R-Yf85kxU5wT4EsKWIyxUMwTyw04QoY,97375
flax/linen/normalization.py,sha256=iaZWE9tL5joqt-8KFfin6_RSPyXD3if9VvEFKl_BFt0,38663
flax/linen/partitioning.py,sha256=KYckTwEimZsKHFdLKBCE3tc42-4D0ANDd9PfoaZZY28,19344
flax/linen/pooling.py,sha256=UjPj4AAoGnMWX4cfgfL8FQ7YAuGUXmWb6AD9Mw1JVU0,5509
flax/linen/recurrent.py,sha256=APbaszIILFlHRkmnawimdnpLIO2Hj-GYIk8XIQ51JFg,40408
flax/linen/spmd.py,sha256=8hx8Pq6afcPKV4laCOs4LuksB129ZFR8AxBWq2V9n3E,11354
flax/linen/stochastic.py,sha256=e8vdlaT_fiRGgQogCZpfTvWdQdx-DignEQVbPhGiJ8I,3356
flax/linen/summary.py,sha256=WZPKWxZs-lVa3JUb_nJLH4WJ3VDDAAMADjjQn-gvj0c,26204
flax/linen/transforms.py,sha256=_mZ7lgPIJbfdmST-ZGTdYh_yuzp0lpe3NN6POiK0AhM,59727
flax/linen/experimental/layers_with_named_axes.py,sha256=XcZfsFKDgRSbVeUZQRt5PGmejpEf23PT3n-AktkAqmc,11754
flax/metrics/__init__.py,sha256=qEgClvMWm4hzoAPfpJiej6gXqpBZ8b8bPZrLB1BRUU0,1163
flax/metrics/tensorboard.py,sha256=HrVu4dUay4XmFDbjFvuwk0IKIqC6Ta11G1e9RHRJthY,7804
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=hf_e_s--4POJ2AoDXd2tMJTcoyodf6Y91T67HLcHfFQ,43050
flax/training/common_utils.py,sha256=0CFVdxcb8TzhbRE_oa8_rQqOw6hEPS8T0xLhkKkxiWE,3692
flax/training/dynamic_scale.py,sha256=5fe_WGIEo94dYGuWQHySkR_F5buKIed0IP9VZh2cNQw,6068
flax/training/early_stopping.py,sha256=OLAL9dO_Z1JrFFL-TL9Gsa0Ci6ECQ3_fOcNmf8qsrWo,2808
flax/training/lr_schedule.py,sha256=rU1e-QFBMAKuXS0HSAMokoavHym2TLoi8y_oDVGgCFQ,7429
flax/training/orbax_utils.py,sha256=_VeOFCbPa1LLiEMeLpQqHJUAcwF7MFAdLIWj5B8_nuA,3712
flax/training/prefetch_iterator.py,sha256=3Pi-n1p_ioCT3t_ZZDLhQWwfg49BMy4MAtzYuUCEqX0,2984
flax/training/train_state.py,sha256=E_bTs1J_SvVM2dH8GvMlETd1ycciveQU05p12sv3hes,4127
flax-0.7.5.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.7.5.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.7.5.dist-info/METADATA,sha256=sPgr1B0TQeB5hANh9EizBPPtoMDC-hCLXNOMZChS2j4,10213
flax-0.7.5.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
flax-0.7.5.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.7.5.dist-info/RECORD,,
