flax/__init__.py,sha256=rdMmFz7H-NUM54KBz4vTh0Eb8IccznXWiit1J2BLHeA,1135
flax/configurations.py,sha256=B0oZ2JQA-5JZ_SO1VyCm_zFD-JwvDmFzH-e5PoSBHs8,5980
flax/cursor.py,sha256=76hSxZHDy5x0T5og8Sz5MOoAcDVS7pwbUORn8DxEzCI,26394
flax/errors.py,sha256=RV-pZOjsEtA5-yTHf4YBe-M2sx48BdYc2v9CIkSwtfU,31437
flax/ids.py,sha256=sbaPq1x8I1hGtZH4VVBufB9mKjfjvhUmwlOks4pN_O4,1752
flax/io.py,sha256=GYViY0BRwzN2i0T6TQ65eJQ8Br7Ye81udlruJXtkJ2Q,5309
flax/jax_utils.py,sha256=FTtXCqYKRoxGTAeLLIYMGn0JFNAph6_LiTl5_cLpL0c,11534
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=EKLEl2OD0mrLNc7iriYZuLwNTuuMoeMMOy1MtpP8oVI,14305
flax/struct.py,sha256=WrY2ChQux3bXpExs6f2Dh8ap_3HOQMoKKX3rv_hUrjU,8339
flax/traceback_util.py,sha256=VSBwID-DMPnPfXlESEkNlqZIVj009mF0kkdlvho8qXE,1990
flax/traverse_util.py,sha256=mTTx6l1r69-o36n_Qx4eenxDWoNtxjkT2Bm_j1fGrIc,13773
flax/typing.py,sha256=sJzy55WA3ztzj91BNkirS_BZKCXM9ovRRjmy6XQzCE0,3506
flax/version.py,sha256=CkZaSEGqjUes1i7QDU4wZ_78oRjjm8DTnSDw9w9t7kI,650
flax/core/__init__.py,sha256=9gthSYfQPBSZvkUNAF6Rl6TzkUJqrEMcL1h4HqPQPWA,1467
flax/core/axes_scan.py,sha256=O98W3DELgcQzTTkxtzCo72FSwYt3YJLMnmG6hht_Z9E,5787
flax/core/frozen_dict.py,sha256=kfwjX141Msh47IxbeX3zjVagGYWjSeMJmuhQrap7AK4,9945
flax/core/lift.py,sha256=fuqxqrkaB8yvkzSfeexJ-KVwIS9dkXUuWcsq4AwCW38,62303
flax/core/meta.py,sha256=7Nvw5SrZfPMXmxgKWP5RhZxQSNCC3wdLboomKA6_390,11730
flax/core/partial_eval.py,sha256=hj_llMsUSSqH1eF6bUdnhWsWFpLlApOPj6Y9d7VbnOg,2546
flax/core/scope.py,sha256=mRCqNWB2s1PdQt1Kiy39dP07Rtjp1KxSoK_O8bobNAE,38308
flax/core/tracers.py,sha256=wzi47_1VH5SbozYVfnvtUfntbrOe2JDlKEjBM0wOwkg,1054
flax/core/variables.py,sha256=KAkQwMGEY-fASlJPHcGpk7vA-8B7VBMlasAus-QfA74,1558
flax/core/nn/__init__.py,sha256=2YpgoLw5eP4npw6MOhgUdfMI4nrGIKfC-fsnDNuULJY,1795
flax/core/nn/attention.py,sha256=9BvYRyJ1Gd3BFBeWQdq5xbY7p5qAyMACl6gcaPL01ic,17946
flax/core/nn/linear.py,sha256=iuOteK4JoyDCQ94O3oeC0Uo8yyxJ84IRurHRU7h-_z4,12040
flax/core/nn/normalization.py,sha256=z8LEGKyWitdq73S_j1WlTa7P5uNJSGL-Qm_eulrqDfk,6897
flax/core/nn/stochastic.py,sha256=svzeqhl0XBLKP2t7Ewgc739xuItFBZvnk7oGjlZSpfo,1488
flax/experimental/__init__.py,sha256=hkK7vneGSpZel3JlyAuXLe-tggix1dk6NsSYwhQXOdE,582
flax/experimental/nnx.py,sha256=YMLSuRAPHEJLdILaiU8_k7rgZ8KKadeYHmoWCq8ux3I,731
flax/linen/__init__.py,sha256=m2nF_aNn4Gz2QADocXDR36JHnMxTg-02Gtx4CWt04c4,5357
flax/linen/activation.py,sha256=LeFjjROwX7GLlyfdA-iBaRNY_qp1PKhGnir4ggxAVV0,4101
flax/linen/attention.py,sha256=AK7ul4A7bxSSQgkVFpctx2B86qZXuyvR-lCHCJ1SOM8,35880
flax/linen/batch_apply.py,sha256=7B_bwfI2vJyt2UPMtOcjEo6puLFavcJeLrF-COoAntc,4172
flax/linen/combinators.py,sha256=vpxANPIeCAaEjQMjQYk-cDxNIvlQJbj6zuQu7yHVHcA,3867
flax/linen/dtypes.py,sha256=0OoAEFJDkwYulIuYkNmOd2a5jJZRi6uap3_EJWXaPB4,3841
flax/linen/fp8_ops.py,sha256=gzDg9faBZ-X-PxxvRM8in7lTdird9F_vZA35IZSDEdw,10644
flax/linen/initializers.py,sha256=CC4lY4uvAGisaxiiIAE2AHVPuN9q8EqQ2e9OZBDPi1o,2676
flax/linen/kw_only_dataclasses.py,sha256=U4-mGzgAhz902A1SL29J8Hdr33smxl8NUB9BwOz-4GU,8828
flax/linen/linear.py,sha256=OnW1-aba0pPo_uRviqPIznr0c73OKjjwx-l_RV8mv_E,44083
flax/linen/module.py,sha256=jo1kg64IZ6gvJCuMTKhxQOEgjffdoHtENPZ6a4SoBn4,116089
flax/linen/normalization.py,sha256=luVKxePv-naECTLyKIpJjpQeRj1C70_Szekh5Ek3zUo,49675
flax/linen/partitioning.py,sha256=p5iJtgfMqJXwON5oQlhnrXqERmN6XR9obMBIr5YrARY,19421
flax/linen/pooling.py,sha256=HNrjOv_ul5QEz27nXg69iSIzfRa42c2cbZh-q8Dciog,5567
flax/linen/recurrent.py,sha256=bwAjMXHK6Htd0BOd39CSDsytolHGWxpmfBIVa30uliQ,45835
flax/linen/spmd.py,sha256=mfYLCV1jeND3RgUG_YVlH2rhmRD3ZHYwR2BDGw6XcdE,11302
flax/linen/stochastic.py,sha256=3S9-YdqrkJArUvPUFrvR2iOYXrYhlZfZPgasOL4NHAc,3624
flax/linen/summary.py,sha256=jte_0klIgJewuGBylLfrcXUe4ihhQCtJbftGlWwil00,26042
flax/linen/transforms.py,sha256=2DK0lGHNQEMKMVDekNAFGdp1_BhWjEdzNW2q6O4dJhA,79634
flax/linen/experimental/layers_with_named_axes.py,sha256=b6cWMTFuT9COOFr7t16Eq4Y5nf6fbqHNksRjvd9lUzc,11231
flax/metrics/__init__.py,sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg,1163
flax/metrics/tensorboard.py,sha256=HUDQYfdhZXTCCxQJMpr9aUzrqexGZasEGvzu9ccy-ug,7792
flax/nnx/__init__.py,sha256=YXcx3YOY90TzBsvT3MyCvKjMy7O_l6uN3Npgxz6mm5g,7944
flax/nnx/examples/gemma/helpers.py,sha256=PliHsbzIxVeN9H-EccyeTO19aG832CHGaRdWypX5A-M,3241
flax/nnx/examples/gemma/helpers_test.py,sha256=8Ssk0kGXScSdTx4iOE1RVUOH_VSwduZ6jOzA1y0BBKI,5099
flax/nnx/examples/gemma/layers.py,sha256=ck2fcJc20vJ0FVrbZtMpebnqca4FUx7bi8pqgMD7mc8,2690
flax/nnx/examples/gemma/layers_test.py,sha256=CAHYMG5ZL2V9ivUKxM4qrvzNbgVvT7lkKyRr4s1gDf4,2733
flax/nnx/examples/gemma/modules.py,sha256=uaWbyHjFR7Y1zg8phGQWeMzKScwpg5CHE6UIDr3x8Ns,10079
flax/nnx/examples/gemma/modules_test.py,sha256=zhohqVZhngW14Bfsift_zugiNnsnMUMNvDI_dxKUfJY,11253
flax/nnx/examples/gemma/params.py,sha256=6lT3fLRdJCR_xhhIIjC_h08kEKSCRcKjvLTbDf11pcc,3156
flax/nnx/examples/gemma/positional_embeddings.py,sha256=Jf3kIBYRKzzowhkwvw-ONZ7wroIG-Z1cdxTm3KpDMG8,2902
flax/nnx/examples/gemma/positional_embeddings_test.py,sha256=uANs2a-kxXE0D_H_Ktu_lbDHqtSR3ecasR54-nMVnGw,3295
flax/nnx/examples/gemma/sampler.py,sha256=7ZyMhMhAwwuUO9Z6d6vQspSQLjPpr_J4f4gESqpMbbA,12250
flax/nnx/examples/gemma/sampler_test.py,sha256=Zh242aS7SsYfnAt-Da0AbX0klgj6FLmInsOi2plib40,10290
flax/nnx/examples/gemma/transformer.py,sha256=PBUE964pncP6JtPOd9EFF_YLwr9a399XfybKu1L1XRQ,10179
flax/nnx/examples/gemma/transformer_test.py,sha256=guTT0_XupayI7cRbkPBEuMSZgEOwW_A3AblyqRKXJ_Y,9081
flax/nnx/examples/lm1b/input_pipeline.py,sha256=H3Nn2Doy5WuDPURjWDd3IGZ5Sy9bZL1M2_xq_3NvEaw,12282
flax/nnx/examples/lm1b/input_pipeline_test.py,sha256=o8sxG9thoPot1LRx4wNbmli7IUfzUsQIYRh-j3riWd4,3295
flax/nnx/examples/lm1b/main.py,sha256=k28k3dKrQQVeQHYoJ8Yi-XyiJDNGz9r10uDTUaCq3Es,2143
flax/nnx/examples/lm1b/models.py,sha256=7lAV8F7dfjCXDgrKhWp8q3O6lNgZUklZX3lcMml9o8g,14993
flax/nnx/examples/lm1b/models_test.py,sha256=l6dM7VumF1m6_HlsdVzyrNqNVSW5g9lX7NUMxcmYyfA,9789
flax/nnx/examples/lm1b/temperature_sampler.py,sha256=tyoiOf4dwoLqkau4tOp7hilU2ocmafSDPYhFeo7oFX4,4799
flax/nnx/examples/lm1b/temperature_sampler_test.py,sha256=ATe15vuNrU51PlJVdablJ60Xn4hkOptUtrCgye5PaR0,1448
flax/nnx/examples/lm1b/tokenizer.py,sha256=djKaOgIuYSL76xVfn7SVDiQdyYaLl5nnYu6LC2M_2Ag,5277
flax/nnx/examples/lm1b/train.py,sha256=8QZrExGYIiXHpFVwtUmCvLpTb0QoBvtIbNUI_Xwq59A,20374
flax/nnx/examples/lm1b/train_test.py,sha256=Ob8Cj6GM3KRTbX4A9hCErb0oXIsKrSXwjaMa1WWTrWg,2152
flax/nnx/examples/lm1b/utils.py,sha256=1rYfFdwXwKDE0ARTDLw3uHU043vi5tXHukzGpl-29Wg,5000
flax/nnx/examples/lm1b/configs/default.py,sha256=rp8zsTa4a4WK_29M57cletqNk6eDsT6xYmCuML24zJ0,5096
flax/nnx/examples/toy_examples/01_functional_api.py,sha256=kB9ajNwjpES9J6ibUwxDt7h56Ajr0SMlmzrGU30k4Mw,2845
flax/nnx/examples/toy_examples/02_lifted_transforms.py,sha256=d8c4VwTEEmHSTb8tVGrwNvz1hYG4tyQCYjNz5SVqJTk,2498
flax/nnx/examples/toy_examples/05_vae.py,sha256=EAVnq2BlgoIzmu8oYnp9kbBlT8KCGTtctQR2SGV1PWw,5195
flax/nnx/examples/toy_examples/06_scan_over_layers.py,sha256=JsOyvnG2datctp5WGK9HYDH09nckq2f-hm42iflYq7Y,1882
flax/nnx/examples/toy_examples/08_save_load_checkpoints.py,sha256=_HMle3KjqrJj0waSoq7X4Hs4jK4A4pN-YgrhU_htP40,1933
flax/nnx/examples/toy_examples/09_parameter_surgery.py,sha256=gc8BbPYuTTuA8qjbRue8--65d9u0bHTC1ySopemHp-0,1932
flax/nnx/nnx/__init__.py,sha256=-m7XZXocHY08GhOmelcdnOjWQ6Szo6WR2HMNnBjdNLU,1163
flax/nnx/nnx/errors.py,sha256=5FCT85_hNWEiikufgReDuwN_uC_4HxZj-VMHS7mIcOY,626
flax/nnx/nnx/extract.py,sha256=LywNen0eKgwnbNMv2Tl3HVg2KVscprzukEICT1hjCYc,12083
flax/nnx/nnx/filterlib.py,sha256=S1Kw7G9eQyO3jRU9pbAyGzNkEYjP3ZMAOJeD-d31HnE,4318
flax/nnx/nnx/graph.py,sha256=QhZ72X15mrpeFelgHl7mtymeZ95JIJKbb6Wk_XhpKeA,57431
flax/nnx/nnx/helpers.py,sha256=KuaUEya-QTcxk_Y415tkLRtvBHzT9QFDVm24cKV03jQ,6563
flax/nnx/nnx/ids.py,sha256=zxKebrldNZvWnyBbgbTvYjL1QiJrJ5zoVQ9GyfuctzI,2325
flax/nnx/nnx/module.py,sha256=ORP40nVOXFMTSbxHUz3sWf0I94-BY5l-cAjW9pIpqso,16441
flax/nnx/nnx/object.py,sha256=FJNFhZS_br0yp95zt04jtmmLNxAll4sHZF19tf8u1kI,6486
flax/nnx/nnx/proxy_caller.py,sha256=2H8il_7cNEbjSymgDnqjggufjRrKlb-KzNg63rmQ9HA,2943
flax/nnx/nnx/reprlib.py,sha256=V6g6PecKngBkpRJgRUXeO9deuYDc2rORJhjan7A3jK0,3178
flax/nnx/nnx/rnglib.py,sha256=Mc3IpevG05UshlxqMijVPDSUgNNtgjdSudJSTGpfVlo,15994
flax/nnx/nnx/spmd.py,sha256=v3F-qvNtCfC47DDiGx62nN16vuSXS1FI9WOXkjcbKA4,6164
flax/nnx/nnx/state.py,sha256=ta_taQX74G3CmxwLtBJkeI6s_iugljxqoeBYqgqlIYE,12363
flax/nnx/nnx/tracers.py,sha256=Pu69niD-sBeE0i_h4OT4vVEkLgCoWIdcV8WCZREM480,1714
flax/nnx/nnx/traversals.py,sha256=Dn0KrPjj9cu-4lUIo-3H2sjEylR7QDUuMK31CColZ4M,4994
flax/nnx/nnx/variables.py,sha256=zchZnkZGsVniitPR6i0SWJ2AmcS4bgZb0xBZQV6g9sc,30264
flax/nnx/nnx/visualization.py,sha256=15yexMZhrxYPnb70m52Yj9I6WC8stxkMsVFx8yoJ4vU,1261
flax/nnx/nnx/bridge/__init__.py,sha256=jKwT8eiLdTn1coreS9I8ApOzJm92DHFUP-udUniG_Vo,998
flax/nnx/nnx/bridge/module.py,sha256=ueboKI0JpKufbRR7Iz-HwSw1LSzfQzl2TEC7w3PQ1ro,6172
flax/nnx/nnx/bridge/wrappers.py,sha256=J46DbagTsfGu5qyvah65X_6QL1USa9zj8-cQUbj8of0,9531
flax/nnx/nnx/nn/__init__.py,sha256=-m7XZXocHY08GhOmelcdnOjWQ6Szo6WR2HMNnBjdNLU,1163
flax/nnx/nnx/nn/activations.py,sha256=Ub94GPKCDh4S4ejqP3r8y3Znh9To78APHCs5AUB-K8g,1219
flax/nnx/nnx/nn/attention.py,sha256=FIEcw4XEBzlZxqCcn5EXLLY6GwNV_SoOJAaXl0N1oLU,24656
flax/nnx/nnx/nn/dtypes.py,sha256=Lpsj2tzufVerQ6HLOFhfJgCwKncHOQqLvw25UPRtVxQ,3293
flax/nnx/nnx/nn/initializers.py,sha256=0bfgv2mMU6iROZwzoyHOJbYpCm4--mjp700rJ5er1dU,2570
flax/nnx/nnx/nn/linear.py,sha256=qhUPRVM7PF9d1DFooqfYeBab6bXXXR3XkjFjA64hFS4,41364
flax/nnx/nnx/nn/lora.py,sha256=eZ_iLa4A0X7Cu8bna84q-fT0YfwOUkjv_3pXj6d9YtU,5977
flax/nnx/nnx/nn/normalization.py,sha256=2mrcXN1KecTe_dqyZYxcq1Uo0fDqpUK6HfHexbW4BGY,28829
flax/nnx/nnx/nn/stochastic.py,sha256=wAJerj8nl2GW31n-ltT4gDZ7oiBuahDzK_paAYqWBfs,4804
flax/nnx/nnx/training/__init__.py,sha256=hkK7vneGSpZel3JlyAuXLe-tggix1dk6NsSYwhQXOdE,582
flax/nnx/nnx/training/metrics.py,sha256=EHAAFrH-_m0IpRMYOJ9p1DmLaVrhJuiOagerhq4xQy0,13905
flax/nnx/nnx/training/optimizer.py,sha256=_OORurPNSw9hPclfJVDNxoIgnssfpTA7uu5dmwuwTtc,7195
flax/nnx/nnx/transforms/__init__.py,sha256=p5E-sUIN2_cMg_e4peF82PzTNbZN7SV556_vMJeVz3E,606
flax/nnx/nnx/transforms/autodiff.py,sha256=0oipdzSQOBd-BG44N-ftaMUq70PyUZfe8JPubcubkSY,24871
flax/nnx/nnx/transforms/compilation.py,sha256=R3msobe9eFlcm8a2Fz3zbZGgYwKW0fgP9ipw67Fz-EY,13699
flax/nnx/nnx/transforms/deprecated.py,sha256=eLlxl_JP7wnYATSKuy_gFUS7lrGWd9na-mUmPc_MACI,57423
flax/nnx/nnx/transforms/general.py,sha256=cG7sa6V8KtIgUH1tPmtIEPueD5oLd34p8_0VL3D4mV0,7476
flax/nnx/nnx/transforms/iteration.py,sha256=B0KGNEt2A_STdI5VGMAdlWv14NVuXlUIjQKJUKA8n1c,35126
flax/nnx/nnx/transforms/transforms.py,sha256=Ypd8lveuSj_5QuekJbZE7YfgsUm9jEWraEqPLUknVdI,4834
flax/nnx/tests/__init__.py,sha256=-m7XZXocHY08GhOmelcdnOjWQ6Szo6WR2HMNnBjdNLU,1163
flax/nnx/tests/containers_test.py,sha256=UAmG1SZDHZ4VFB7DPr0EC0BCOp4AcM4NqfHpf9UOoo8,1723
flax/nnx/tests/deprecated_transforms_test.py,sha256=wgByc9rG5-poFXE4e1u0jIyowMZjhHr4mS9N85EeAUk,10069
flax/nnx/tests/experimental_test.py,sha256=ooAK-thQWUHb3UVJMbbQGyP0Hc4aDjMcarkRiXcJZ40,664
flax/nnx/tests/filters_test.py,sha256=1Rk2CPBf6LAP1c0puFFXtfLqwJSi5VnlnxoG70FzKvQ,1049
flax/nnx/tests/graph_utils_test.py,sha256=icEm84mikupJv0K9gXdxMj0e97bUsaNfjHaGrnxbkcg,23568
flax/nnx/tests/helpers_test.py,sha256=O5tQZWuAOclEPG3uNBNt0O8d5ZFJgJwlfzXA8jYdDH4,3154
flax/nnx/tests/ids_test.py,sha256=RHmAsrQ1GqiTrlJNhMtTaPVE9Yx2fU37E81ynOZL668,997
flax/nnx/tests/integration_test.py,sha256=ukkaYliMq_ifNvzdAIgeHw0xnCyCE7lTFtqiU498wuk,7494
flax/nnx/tests/metrics_test.py,sha256=zU9eeA0IriP-UrbbQrxpwZV8sQ0qUbMgusv5qp6Z6cc,4288
flax/nnx/tests/module_test.py,sha256=Qjcc13dXr-z-XWxIv-mMsgDVo7ZeUdqNeHONJ5yOYcQ,17280
flax/nnx/tests/optimizer_test.py,sha256=MecQfSjHJrkZq_PcufdUhc8K8wUjJt0Zjy9PBcUdwKA,5639
flax/nnx/tests/partitioning_test.py,sha256=0rhHqVX189EqyLXEMdjN-KxkT40OFVFqJ1O7jgz0zI0,4235
flax/nnx/tests/rngs_test.py,sha256=IO-kTlDSHPHaYOHU3pcRlGt3WvZPwvwzRaQjghIIsNk,7764
flax/nnx/tests/spmd_test.py,sha256=5hg4Qbyc-meRePgYI_dCTC07KfW_NdXt3CHF0wAbiP0,2947
flax/nnx/tests/state_test.py,sha256=hDl-eRe2ghb8Hy3DCdszxgkCwLuTp2r2Wwcg6tFNoGs,2478
flax/nnx/tests/test_traversals.py,sha256=phooI5tvVrq2jwmEqWhyKRR50W81sZrkIzNEcZ8Pu0c,2749
flax/nnx/tests/transforms_test.py,sha256=8IUo9Ef__W-A03tqkBsR-1p5QAd-uWUbunG0lA_yVQY,64175
flax/nnx/tests/variable_test.py,sha256=JU8LY8_Qo7XF6gUHpQng4nPwXTty7-fxPUZsdRije_Y,1939
flax/nnx/tests/bridge/module_test.py,sha256=pzA7BvHM61saV3w3ihKMzwBcxR9QWtILXWDyzP1kfE0,3396
flax/nnx/tests/bridge/wrappers_test.py,sha256=g5xLHKdN-nF6LD4FNAMBj8rSayU7nnNVYFyN2UtrFPw,10310
flax/nnx/tests/nn/attention_test.py,sha256=AJkS5vOGDAN807Jw4sPztDvd2eKByPl4plIUi6qOqro,5002
flax/nnx/tests/nn/conv_test.py,sha256=uSAV7VLsbZBxnbX_MnvSVYWpllfCFvWUOTJ1HZxIlos,5162
flax/nnx/tests/nn/embed_test.py,sha256=HYBHbxxTQjwWqvM_ZIRtZsM-2xV9OlVzyNy49BmOo1k,2278
flax/nnx/tests/nn/linear_test.py,sha256=L3CgDiVoej1meMHc0q4kkA68js-526eLBr2gRDuwOuc,4559
flax/nnx/tests/nn/lora_test.py,sha256=BanXMdVaCakb7_7eBuIB0L5wTiQUQ4hNRm_IoM9YMW0,4230
flax/nnx/tests/nn/normalization_test.py,sha256=jOhkhEKr4iVG6fhZEtCETpUs-tI8DYCcDZ3xsq3qO98,9997
flax/nnx/tests/nn/stochastic_test.py,sha256=ZXDyZnmLuE5oMVO6FsDMh91Ch_kIl4l3bb_8u8Y4YzU,2596
flax/testing/__init__.py,sha256=UarXpjywrTaGYJBYPf0joEzJawWavRs_f58YZJowfh0,647
flax/testing/benchmark.py,sha256=7dBf47nNxYbu9HTko5yR13iw5TyEip8FKaTJ4FE3pI8,9286
flax/training/__init__.py,sha256=uJTipyYX83XazXl5bkf17EF4DjJE5s1YhCA3vS0vXqo,613
flax/training/checkpoints.py,sha256=mT4wVCc7w8SbttIk4qT5USlT5D2f3QFE_P-pXg2XsCw,43779
flax/training/common_utils.py,sha256=_3-gAQuFBC6ITUiu9dgGMZu0PS6SLlzd59UlnwfHAJg,3690
flax/training/dynamic_scale.py,sha256=CHgKotSqPKd9idn1SALIJ0xm_WldjhhAPfNR2kEpmF0,6054
flax/training/early_stopping.py,sha256=uy20C4GxzYI7GV0re-vYH-Itj9f_u0pf6M2uZPe8IgA,3194
flax/training/lr_schedule.py,sha256=SYCWUEgl_iZyOZpIFaP2wzIgniamzu6I1MDS1R4Hlsg,7419
flax/training/orbax_utils.py,sha256=LHB60fz7R9bO2zF3l0RlK-YBD3i-NSfPnukld772Q6I,3116
flax/training/prefetch_iterator.py,sha256=KnhVdWbGC--2YTaxKeuGJTfH2TrvtjQWP3mk3s6KHZQ,2978
flax/training/train_state.py,sha256=s5O7ahzTTgzZb-ZICxnDvzJvTQPM9iqcUj3H8xAtbnI,4781
flax-0.9.0.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.9.0.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.9.0.dist-info/METADATA,sha256=wQxyaCmrrN5oQ8AbwnSsus7tjw_h1-3uWrTofUTXKho,11103
flax-0.9.0.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
flax-0.9.0.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.9.0.dist-info/RECORD,,
