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=BCq1Vt5IL1VyxAI_udnOT7m_yb9tIvCldEKQqPYujvo,31634
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=EIhH9xANwfg7qLJHCGcm5Hdysa3mjqCjqwe4tlqsJQ0,651
flax/core/__init__.py,sha256=G2TuucsjMo9GsYkPcGrs_AxxKZ1osNbxU7De9CRfJuY,1436
flax/core/axes_scan.py,sha256=O98W3DELgcQzTTkxtzCo72FSwYt3YJLMnmG6hht_Z9E,5787
flax/core/flax_functional_engine.ipynb,sha256=DGo3J5pSMS6BaDhQKUopSAt6NOQ8dp1lF0v8w63clMo,10482
flax/core/frozen_dict.py,sha256=kfwjX141Msh47IxbeX3zjVagGYWjSeMJmuhQrap7AK4,9945
flax/core/lift.py,sha256=p5OxlfVjNNOt5kMaiGR9OfwPBACt2Wi7gRWa83TSPHs,61910
flax/core/meta.py,sha256=BH0waiaNqIJWFbkEQpy6X5ZL5UWy-815i8N9Iyc8l3Q,12312
flax/core/partial_eval.py,sha256=hj_llMsUSSqH1eF6bUdnhWsWFpLlApOPj6Y9d7VbnOg,2546
flax/core/scope.py,sha256=AJM40uxOTRARmmshLEjDuwt9MCBbiyPGLt-WHtrTNZw,38287
flax/core/tracers.py,sha256=KnfKHqMsc1ePABzyxFX9tFRd77elfmkC0N_YFY7Ifvg,1071
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/README.md,sha256=fQOTJDyHjMuIUiBxOomju528Xfolp3oGBo9q5l9U-Uk,2191
flax/linen/__init__.py,sha256=r21DuMLiLLn1LwIYTIopV8nENDlrbvVZowwBEeOAqZc,4820
flax/linen/activation.py,sha256=LeFjjROwX7GLlyfdA-iBaRNY_qp1PKhGnir4ggxAVV0,4101
flax/linen/attention.py,sha256=AK7ul4A7bxSSQgkVFpctx2B86qZXuyvR-lCHCJ1SOM8,35880
flax/linen/batch_apply.py,sha256=P4mjWNJiPw-RpC5FMJzLeguM-qriMR2x587DRso7cTw,3590
flax/linen/combinators.py,sha256=vpxANPIeCAaEjQMjQYk-cDxNIvlQJbj6zuQu7yHVHcA,3867
flax/linen/dtypes.py,sha256=sc43s9T4GdIBFmQ-nf1LcB4P85UpHMoflQo2DY4dO74,3259
flax/linen/fp8_ops.py,sha256=MGnEEaLp_No8be7eKq6VL7auF5JoJGstBdiRfoiwDeg,18649
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=WTsn6bSb95eCoLMcgeKD9DuHbCNe4-O2u-Gngyw00mg,116846
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=yD_yUt2EbkIDIE0LhEF1gwA97udWCfvYk1SN8isOFrQ,11973
flax/linen/stochastic.py,sha256=3S9-YdqrkJArUvPUFrvR2iOYXrYhlZfZPgasOL4NHAc,3624
flax/linen/summary.py,sha256=MGXYpqY_7_CuZs8eJW5Ly59clnQLNXnV0xkCP79TC-A,26164
flax/linen/transforms.py,sha256=oMBVjiS-LoNlbIwxc3qzVbYJcaQuwcQn93oIU18dUtg,79151
flax/linen/experimental/layers_with_named_axes.py,sha256=b6cWMTFuT9COOFr7t16Eq4Y5nf6fbqHNksRjvd9lUzc,11231
flax/metrics/__init__.py,sha256=XCxb3apNmuctXSifbNcZMwlDQ-moSYF1SBxg40Y0UCY,581
flax/metrics/tensorboard.py,sha256=HUDQYfdhZXTCCxQJMpr9aUzrqexGZasEGvzu9ccy-ug,7792
flax/nnx/README.md,sha256=6P7ujPddeltA2tBEvF1tPIfiRtJw185FWylFpGW8rnU,3385
flax/nnx/__init__.py,sha256=Z6y1QVI5Sj_ZYFpxEx1l2jl3f_yyy5XdRNygxQ3MmKQ,7542
flax/nnx/extract.py,sha256=Vc_FMFiefclwtMrM-DJreBMUS7mUIudtKlfPDFrgKzg,11467
flax/nnx/filterlib.py,sha256=q51cxz-8irKzU_g3Hg_HLPqfTEx34BBXi5zjAB2MxvI,5217
flax/nnx/graph.py,sha256=F66CaC2tGU7BqFd4x3SXT3ll6zKZrfUCHsjrpdtRdaY,54723
flax/nnx/helpers.py,sha256=vVHsFMoqclXG3x72UBZWLNR_qUmPg_2XJzCW6jJGMn8,4569
flax/nnx/ids.py,sha256=mAFUaRQrB89E0fg1R2bzehpL09HOQSyceoRwF5RDC_E,1743
flax/nnx/module.py,sha256=l3xRIaZvw4uYl2ikzQwlcI6ZH2BLwj21L-mrd_QCE0s,16422
flax/nnx/object.py,sha256=afbzXVCugcFQS3dbAMqCC2VoN_baJN2-O2g8EYCpxzY,6492
flax/nnx/proxy_caller.py,sha256=UqSWnOCB2anBRQgeEFZ9zAUOnPXSJCoBCp9PKoPpcdg,2361
flax/nnx/reprlib.py,sha256=V6g6PecKngBkpRJgRUXeO9deuYDc2rORJhjan7A3jK0,3178
flax/nnx/rnglib.py,sha256=lw7rvbOqmVPQBl-egz_CeEBAd8SE-72hBz_jh86kqZA,15531
flax/nnx/spmd.py,sha256=Sa40FjOOKMURYkRiEUx9k6OPkvdYWbFq2dQQz7xfHA4,5271
flax/nnx/statelib.py,sha256=gSAsNSWeN2V-wedu1CEVrCTP7srhClSkxAmkWGy7Ppk,13569
flax/nnx/tracers.py,sha256=0zKB0HelO7MqA8HFoNWr0NeapzMjIVZqIcSCJaZX__4,2140
flax/nnx/traversals.py,sha256=Dn0KrPjj9cu-4lUIo-3H2sjEylR7QDUuMK31CColZ4M,4994
flax/nnx/variablelib.py,sha256=3ub2t2LUe9I8309tMuLWiR4xhYrJ-u6BtIdi3cjsC2I,28418
flax/nnx/visualization.py,sha256=15yexMZhrxYPnb70m52Yj9I6WC8stxkMsVFx8yoJ4vU,1261
flax/nnx/bridge/__init__.py,sha256=yF_wLQys8TdOp7ROJy2fEnyA6JGg2voijB4DwO_cm0s,1132
flax/nnx/bridge/module.py,sha256=PvrL5rzNCbSlgmGej3gMtLep2311xarhHHKSFMXEWew,6156
flax/nnx/bridge/variables.py,sha256=sSO8W9Vuxbb6MgxoMBLUh7PlLSyt9LHSqcolG7s6DfM,7690
flax/nnx/bridge/wrappers.py,sha256=OCukbwDlWLgdfmqvLGd3XmWSVBzjBQOxJDBqB6RMvCU,10118
flax/nnx/nn/__init__.py,sha256=ECrkYim1U12kMsc_1l0r-6drbITTHaO64JB5LZCuV7A,581
flax/nnx/nn/activations.py,sha256=Ub94GPKCDh4S4ejqP3r8y3Znh9To78APHCs5AUB-K8g,1219
flax/nnx/nn/attention.py,sha256=2dG3ewX_mT9kcfkuUG1K3ty8c-UE7sSYH0H7kfuxG7Y,24632
flax/nnx/nn/dtypes.py,sha256=Lpsj2tzufVerQ6HLOFhfJgCwKncHOQqLvw25UPRtVxQ,3293
flax/nnx/nn/initializers.py,sha256=0bfgv2mMU6iROZwzoyHOJbYpCm4--mjp700rJ5er1dU,2570
flax/nnx/nn/linear.py,sha256=Ip-CigjZBdoTEslInXSwt2YwnqjXS4x6lUqWR6dfmE8,40823
flax/nnx/nn/lora.py,sha256=pEmfrFJ5rvAcmWEF4IKG1OYFKulWO2iNHkrzrO_8egs,5387
flax/nnx/nn/normalization.py,sha256=lnxzWPzTDasdXwhtH_ZeEvHCjeCRiMsQWTV64GKU2rQ,28817
flax/nnx/nn/recurrent.py,sha256=PrRxGPFB3q-Uj8DNj4WwNvYXxVtGa-DXY65se7gskv4,31017
flax/nnx/nn/stochastic.py,sha256=XJge_0NtC6CQLLP7YNWBDfAh4b-DuhemNMXwwdVu6r4,4214
flax/nnx/scripts/requirements.txt,sha256=xxQ4qS5Xhzyxa4FDTLP4vvIvM8EStbTyLWpAMwKLH9o,17
flax/nnx/scripts/run-all-examples.bash,sha256=pA5Cm6xK0Gj16hDSZMBchpq7rtw6n3K5ifcBqXlwAac,260
flax/nnx/training/__init__.py,sha256=hkK7vneGSpZel3JlyAuXLe-tggix1dk6NsSYwhQXOdE,582
flax/nnx/training/metrics.py,sha256=F7Qf6wFNOIpOX_5oM8ACeOdEmT6SBpy1Ui00tv9zNhc,13313
flax/nnx/training/optimizer.py,sha256=8mic2vLXY3jYDF3IWt6W89rocm7sXbGM-krnJAjpdO4,8430
flax/nnx/transforms/__init__.py,sha256=p5E-sUIN2_cMg_e4peF82PzTNbZN7SV556_vMJeVz3E,606
flax/nnx/transforms/autodiff.py,sha256=dAxIZ_FbJ9h5orNW0CKgAegfZ7t7I4uAVss-pACTucw,24890
flax/nnx/transforms/compilation.py,sha256=TAKzQTN4UVvvRHU8p2mIuhdt7ZTl1A1HOV4xdlA62Zk,13718
flax/nnx/transforms/deprecated.py,sha256=bTgA_X6aKyehD4knudgrik2Je-CLULtn025GKaQTRZU,57172
flax/nnx/transforms/general.py,sha256=V-RQtCG4YBcTKrm-nLYkz1E07G-X0U4KHfvgU9mYYjM,6890
flax/nnx/transforms/iteration.py,sha256=nL2gygXbDlTZNcwKrA5kcEWv41VZSOHHzr6UIyikLVQ,40937
flax/nnx/transforms/transforms.py,sha256=Ip7DPbfD-wG5_88NbvFNgN1o7SYdWzXFSmeipJrUIRY,4236
flax/oss/ .git-blame-ignore-revs,sha256=eJmLQj4P4rVEN8uRkflxwOhLFte29Y0WeGoI1uRNWfY,443
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=hy_rTqe4aylHtOqVX5Pp-8T0egbv-CIWbopaFdg6v30,43820
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
flaxlib/.gitignore,sha256=aEV8LlmiqurG5Q51lo0VV2wYEIM3aOrUKwSqckRJghU,713
flaxlib/Cargo.lock,sha256=gnWzds8mOvuwGpv4eCUmH-1gsS17unTNlIo5GtFxi2E,8092
flaxlib/Cargo.toml,sha256=dXyBBsBJV4QbEwRtdac7kXU0eDs_M-6im6zN4wV38Gw,242
flaxlib/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flaxlib/README.md,sha256=KEjDOflDXcDBheXH9LC9AhH3uDc2bplID-7HzF5fMO4,9
flaxlib/pyproject.toml,sha256=8f1vSp01Bw1CdTIyxgSK9ZdbBZpzOolh1vIkRMNE9mE,446
flaxlib/uv.lock,sha256=Gykoqz2Ml_L223saaWMHknfUC9791NhGPs52iu3AD8c,4842
flaxlib/flaxlib/__init__.py,sha256=iN2Q0A8-vOzM4E_56NXvorOXG5PEY2HJ79TTNKANq3Q,641
flaxlib/flaxlib/flaxlib.pyi,sha256=oEPyZG_mFeRbff798Lc0E2DeiwxuVxkAa-sqLP8HHhU,628
flaxlib/src/lib.rs,sha256=6sNTC7o1YdUKXuaWme49ltvAygky11oM2mkW2TJ2KyY,955
flax-0.10.0.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.10.0.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.10.0.dist-info/METADATA,sha256=SjCb2x4dn2hcf0iSLNBYwifWlUZcN22VhKXocG8IzPc,11224
flax-0.10.0.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
flax-0.10.0.dist-info/top_level.txt,sha256=QHAaaaSiUsGtDTsjBNEKcGlbDu9bQCWS0KY96gbQYmU,13
flax-0.10.0.dist-info/RECORD,,
