flax/__init__.py,sha256=rdMmFz7H-NUM54KBz4vTh0Eb8IccznXWiit1J2BLHeA,1135
flax/configurations.py,sha256=bngv2s5xqu3hpH-odX5EgqZILpUv-Y-qv53fc-Hrq-w,6174
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=qfmExEIrXwD5Jj0vynQ_Cv78lZ_RG_6HIBRn1lrJaOE,11691
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=EKLEl2OD0mrLNc7iriYZuLwNTuuMoeMMOy1MtpP8oVI,14305
flax/struct.py,sha256=691XQuzWZSG1OQPi_DRJJTUYuHIv3yRmZN7zsVaT8ZU,7991
flax/traceback_util.py,sha256=VSBwID-DMPnPfXlESEkNlqZIVj009mF0kkdlvho8qXE,1990
flax/traverse_util.py,sha256=mTTx6l1r69-o36n_Qx4eenxDWoNtxjkT2Bm_j1fGrIc,13773
flax/typing.py,sha256=lvobB1OGwOoa24Kx5gE8FxvDvVX1retalxqZVpvXu8A,5063
flax/version.py,sha256=bTvvzdqjLCGC4Dt-N7I_psfrpbXLOTJg-jy_WCJaWRg,651
flax/core/__init__.py,sha256=G2TuucsjMo9GsYkPcGrs_AxxKZ1osNbxU7De9CRfJuY,1436
flax/core/axes_scan.py,sha256=k4MWcXGUoP4tvh92DtufuChi_RTvwBFw-egALCBApo4,7807
flax/core/flax_functional_engine.ipynb,sha256=DGo3J5pSMS6BaDhQKUopSAt6NOQ8dp1lF0v8w63clMo,10482
flax/core/frozen_dict.py,sha256=kfwjX141Msh47IxbeX3zjVagGYWjSeMJmuhQrap7AK4,9945
flax/core/lift.py,sha256=VLAX7WzUqflg-_q7Per0B_6Ffde96o0XEWPdGgtQHCE,66511
flax/core/meta.py,sha256=ThLozsDaSZcgUp4FE9eV9DVnQncX3kiusRBOa6aXdL8,12564
flax/core/partial_eval.py,sha256=-wEOAjGjV9qbN9v9NCEFCMYrIiDHdRN35WfIRoXD7i4,2650
flax/core/scope.py,sha256=GynS8cqCaFGnoUNbrEBCrnvq5zuZXOqwn_8MjO60C1M,37083
flax/core/spmd.py,sha256=gbhNROjIiYMNswMClk7_BCTrH5saKlC7kEMiBg7yFms,2215
flax/core/tracers.py,sha256=SspU49KWQA5ojai72v4-RQWXXMqVCbZmxPYEom93_X0,1037
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=oF5lZ5ZbcZTe8BZQiHzosohdJxk7ne_O1vR1ZgJLReI,4860
flax/linen/activation.py,sha256=9nP73bYSau_Ab6vjSfrKWZfC1H2iUmy0l4qgosVBRSY,2843
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=lT_IYpOj-VhxQ6Dlco-0NGvl2Yg_8ppxtM-n2f9yVhE,44083
flax/linen/module.py,sha256=5I2TP0eY_OaFb1meDum6z-nhvNTUALI2qoZ9x0vdnWU,117170
flax/linen/normalization.py,sha256=aBtZqpXlukWkmKfG5cmLVZqunL5hIfp2vow8fRoKXqM,50737
flax/linen/partitioning.py,sha256=_nugFM0fKZAgBXvgxEh6EsEfI5tA6vb4k6YrticiqXg,19273
flax/linen/pooling.py,sha256=HNrjOv_ul5QEz27nXg69iSIzfRa42c2cbZh-q8Dciog,5567
flax/linen/recurrent.py,sha256=bwAjMXHK6Htd0BOd39CSDsytolHGWxpmfBIVa30uliQ,45835
flax/linen/spmd.py,sha256=5zmqTPCGDfaT4sLfGQhBt-68XDrt2j-xt55E5oYW5o4,11433
flax/linen/stochastic.py,sha256=bvSVnu__BOvZkLaljmbXYtsNyVuc03zmatWUQ2SuSbg,3628
flax/linen/summary.py,sha256=MGXYpqY_7_CuZs8eJW5Ly59clnQLNXnV0xkCP79TC-A,26164
flax/linen/transforms.py,sha256=_Cu9aXdHfJ8B5bKily6n5HrC7tUjiz-ZaxEKjjItmCM,80581
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=vIILklaExGtlRyhyFity5T2_H0AtBAS_yvLRmQYlpnw,3360
flax/nnx/__init__.py,sha256=vVwbw7_ChINZOvlEQ1vBB7vKN6vPYsQPdw_hIkpzhZw,8547
flax/nnx/extract.py,sha256=uvOXMCqn_TpSzpv8HLIEpToGxporhUHTEEEZmT1dd4U,11022
flax/nnx/filterlib.py,sha256=7ZG8IETSDCOaXdbZeskdJWcBblR0AFQQgNVwVvFBUxI,5336
flax/nnx/graph.py,sha256=umArq_qcQTscrGvcZ3jXPMYu8kzeB_3cy2vCA6Yk7XY,79019
flax/nnx/helpers.py,sha256=WCMLxus4sjzoWm_xvT4pJzI1IGMFHgMixntX02GZwVI,4657
flax/nnx/ids.py,sha256=mAFUaRQrB89E0fg1R2bzehpL09HOQSyceoRwF5RDC_E,1743
flax/nnx/module.py,sha256=i3ShnTcPwohLWyGgPGAMbcw84AlJRTWlUcArF2nnkYU,18825
flax/nnx/object.py,sha256=DKzAby1DszfwkXoDNbCuReAoqDaABpG6fYhvTQ5_CzE,10444
flax/nnx/proxy_caller.py,sha256=UqSWnOCB2anBRQgeEFZ9zAUOnPXSJCoBCp9PKoPpcdg,2361
flax/nnx/reprlib.py,sha256=T8nztZQdJzt32VJCoVm6KyxsaHK-OtMPXRiKvMStZ_c,6823
flax/nnx/rnglib.py,sha256=lyVZHmDAuD6iDymI3rHgMIm6N6SR0ROXrsBTtW5JQpE,15867
flax/nnx/spmd.py,sha256=foKNdDAbxCmqnpaUA0x8hCTFJrqFFCouRz2aSpLUG2I,5595
flax/nnx/statelib.py,sha256=sTsSBdN06LVyrv3Awag-nymGWnGBRSHH-lPP8BY2kn0,22010
flax/nnx/summary.py,sha256=GrIt9r-KW_ZXyJC0i_ukBKr9qs6hAcmxL6DDTV6f0jE,23767
flax/nnx/tracers.py,sha256=smlfeHZQKxFj3_btwmGpsbG9_qUKz-nNM3EpSXMoMN8,2311
flax/nnx/traversals.py,sha256=-4dpgYN5k7vfTXMkzRKdIJoEW4gohht8MLS0BxwIRG4,6341
flax/nnx/variablelib.py,sha256=Wgq7sFg1MTzoSd7Y0zCjaym0JpSOz2tUEtulGpJz9kE,31946
flax/nnx/visualization.py,sha256=344vI1OijvzAi3ZRosg1b3bdzEclu8rJE_pVDkELh4Q,5168
flax/nnx/bridge/__init__.py,sha256=LHwLrUmNytXmNHPlaFEOCmRdeT7IzPzMQZJ1DIpVznY,1057
flax/nnx/bridge/module.py,sha256=CbdMj0VN-tXCOf19yic4zJp8EbaHaeOJxP4RN7Iofrk,14843
flax/nnx/bridge/variables.py,sha256=W4HfywK0_jrrrAWpXn6q7oVB3IqS56AOcHJfnkezJY0,6404
flax/nnx/bridge/wrappers.py,sha256=W0WfBZLrSqR9JDDINFyywMXoz3oS7cf6jLVfiqNQK3w,10441
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=giVQG35NGgTWecgERFrCg6AelpWY1zTV0Nl1_zed1_k,24754
flax/nnx/nn/dtypes.py,sha256=Lpsj2tzufVerQ6HLOFhfJgCwKncHOQqLvw25UPRtVxQ,3293
flax/nnx/nn/initializers.py,sha256=0bfgv2mMU6iROZwzoyHOJbYpCm4--mjp700rJ5er1dU,2570
flax/nnx/nn/linear.py,sha256=p2mGNLADBM8Jagmo2UQYm9kMrg7Uk5053Yw-HYjCp7Q,41000
flax/nnx/nn/lora.py,sha256=M4hYvSPMcKbji_3Md13Tqg8-1D9JXAQ7QeJFqeWbdHc,5911
flax/nnx/nn/normalization.py,sha256=oRgne6wj0MCuiGIS8MY_hN1uVpfORylH_gaRKEeNk20,29241
flax/nnx/nn/recurrent.py,sha256=8x0JWPdoVm3EqvZYo6eNEacJ59xBo_DeCnCevEaVUQ0,30688
flax/nnx/nn/stochastic.py,sha256=vRTE9KZGaECGzjbFFZclgd1f-hvW6BblIaHbQH2IX-w,4216
flax/nnx/scripts/requirements.txt,sha256=xxQ4qS5Xhzyxa4FDTLP4vvIvM8EStbTyLWpAMwKLH9o,17
flax/nnx/scripts/run-all-examples.bash,sha256=UWvylH7QfAlWAB-FVmlNEL2QKfJ1Omp9P4mRxBjGKkI,252
flax/nnx/training/__init__.py,sha256=hkK7vneGSpZel3JlyAuXLe-tggix1dk6NsSYwhQXOdE,582
flax/nnx/training/metrics.py,sha256=Gm2LOZLql7JdRKbvncav1Zr2tUoAOVPzqFGKpmWFQfY,15265
flax/nnx/training/optimizer.py,sha256=6tqsyITtnjVKn0tnyHeFCVJ9t7K_bt0EGwWOYg2evY8,8685
flax/nnx/transforms/__init__.py,sha256=p5E-sUIN2_cMg_e4peF82PzTNbZN7SV556_vMJeVz3E,606
flax/nnx/transforms/autodiff.py,sha256=f0fQxABi6XKhpqV-6JMPY2EEsv26sfLNJ3d9wXgATX4,29041
flax/nnx/transforms/compilation.py,sha256=OfpjjHtJsMEG-FgQfFmE76ku9z35Q1-Erfjw9OjS3ZI,24638
flax/nnx/transforms/deprecated.py,sha256=bfdlhcVBhBaq3Wj5O0kxTccLyy2lVqtxxoerARM3jMw,57354
flax/nnx/transforms/general.py,sha256=F7JQCB_WxN0vv0h2vzbPnzmBE-nUjArS4dOfR6USXso,6933
flax/nnx/transforms/iteration.py,sha256=jfSx2OqYbSrtddAAT_UD_L0a67doHvoHlv6mieD1uKg,48575
flax/nnx/transforms/transforms.py,sha256=-EXE-uQW0hn_LGt7-AqdNogAf0kOgHt7N9J4kagueHI,7131
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=Jm9BtJfyRsPypx_UK_OD19gy_PEPzuFejpg2UmEO5z4,43662
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=2xZDNe0C12dZWpGwtBx1CYvk8_bwnsfpNje2jBR1Ikg,4782
flaxlib_src/.gitignore,sha256=aEV8LlmiqurG5Q51lo0VV2wYEIM3aOrUKwSqckRJghU,713
flaxlib_src/CMakeLists.txt,sha256=IKG6oubL8sTh3DNoWzUDqVFjulxzKrv3HSwe_VlR_Cg,2220
flaxlib_src/Cargo.lock,sha256=gnWzds8mOvuwGpv4eCUmH-1gsS17unTNlIo5GtFxi2E,8092
flaxlib_src/Cargo.toml,sha256=dXyBBsBJV4QbEwRtdac7kXU0eDs_M-6im6zN4wV38Gw,242
flaxlib_src/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flaxlib_src/README.md,sha256=PRENmUKixvRUA1uuY57x9EhwMwfBowPy8nAAUG4QEgA,734
flaxlib_src/pyproject.toml,sha256=aSvyHsXB-rDr9aXNrZNPb6sfL_tHMnqRZNVp316E6YA,705
flaxlib_src/uv.lock,sha256=Gykoqz2Ml_L223saaWMHknfUC9791NhGPs52iu3AD8c,4842
flaxlib_src/src/lib.cc,sha256=MPY7D-_BLlqjGlHQ2vMDI20LzmfKcvS5j8I4upFrwLM,8309
flaxlib_src/src/flaxlib/__init__.py,sha256=D2dLrkD_OhJYtLNEWKmAOYZWLPGLrtfZJlALF2FIeiI,690
flaxlib_src/src/flaxlib/flaxlib_cpp.pyi,sha256=gy0KpecY0i2zsWq9KuzCVqvnJqqB9GhnWjvlqID0arM,794
flax-0.10.4.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.10.4.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.10.4.dist-info/METADATA,sha256=bgrJJ54zhOioQPfiXUPPZ4zosIVT5dBETz_v8cGhnPU,11189
flax-0.10.4.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
flax-0.10.4.dist-info/top_level.txt,sha256=ob6CnPYMaocT-XGMXv3h9nSaKwe-A1SaSn6Qeg2Xfec,17
flax-0.10.4.dist-info/RECORD,,
