flax/__init__.py,sha256=rdMmFz7H-NUM54KBz4vTh0Eb8IccznXWiit1J2BLHeA,1135
flax/configurations.py,sha256=hbJ0ZNNKYFXvebmafQu01IOkWl-iiG219ikiCWjVKc4,8695
flax/cursor.py,sha256=76hSxZHDy5x0T5og8Sz5MOoAcDVS7pwbUORn8DxEzCI,26394
flax/errors.py,sha256=Dz9kBdR5yPF6VaG7FARpe9i2Oafl7zdm232x8cfAVGY,32192
flax/ids.py,sha256=sbaPq1x8I1hGtZH4VVBufB9mKjfjvhUmwlOks4pN_O4,1752
flax/io.py,sha256=GYViY0BRwzN2i0T6TQ65eJQ8Br7Ye81udlruJXtkJ2Q,5309
flax/jax_utils.py,sha256=pTBZnmAc-zJm9lamiZzgh6Khj37u8o1wMYIESY0I1xY,12413
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=EKLEl2OD0mrLNc7iriYZuLwNTuuMoeMMOy1MtpP8oVI,14305
flax/struct.py,sha256=4paUjwwyGtPbZc7ndEG6alJg6RB07NjEvHul4CRU73I,7769
flax/traceback_util.py,sha256=77RUsCJNAGRVOG3aCQIQQkNenl-1ztZ1cmNS59J0Rrg,2075
flax/traverse_util.py,sha256=Q9rFcUQlhaGgDwVZJvBBL4vJKFL1hQ7hEzZ0zikUW8Y,13840
flax/typing.py,sha256=IwriXie9KGh9xXk3KR_ZAlwERuBKny0TI4MjCgPbYIs,6744
flax/version.py,sha256=ImNADWB7F3dGnD6z8xs73dtXy2hbUUOqDwUe0rR9bkA,651
flax/core/__init__.py,sha256=G2TuucsjMo9GsYkPcGrs_AxxKZ1osNbxU7De9CRfJuY,1436
flax/core/axes_scan.py,sha256=S35lGPoHJazISPgHm-8rBKb2gZnPoEMJjOnj0dS4WWM,8592
flax/core/flax_functional_engine.ipynb,sha256=DGo3J5pSMS6BaDhQKUopSAt6NOQ8dp1lF0v8w63clMo,10482
flax/core/frozen_dict.py,sha256=d6LlT3Qwh7QqgvFOA1orCt3Hl06OJ6oWuBIUBQ0oRtQ,10301
flax/core/lift.py,sha256=JVYAc4hNaQCgTgP-6PBhvjQcyDYp_vcxulENZE1O21w,66832
flax/core/meta.py,sha256=nnZ2Ol8vDOhEMvB2IMFidjgk6NgLwN3_OEhTuLPOu7A,12638
flax/core/partial_eval.py,sha256=-wEOAjGjV9qbN9v9NCEFCMYrIiDHdRN35WfIRoXD7i4,2650
flax/core/scope.py,sha256=_a0P0d_qTGibuFVFkxssurip1phgbHNOZd1xOEdPMzo,37289
flax/core/spmd.py,sha256=mTq7vYXL3l-YIxMJjronOk5Vy6YJbMDtXxm2QjgLEOU,3724
flax/core/tracers.py,sha256=W1hk1nJk5yEFtAwu_CaWtttQEGaoBeq_bI9Ig9gGtv8,1156
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=eV-eDG-f5nhH5TZmt7E3t0kCFfMgVzlFfWSLBj9SpiQ,4924
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=0MtnkJfPZ1idowIAp5VUIYpiA_tBMUWWW-0yL7YQlcQ,3298
flax/linen/fp8_ops.py,sha256=ypHenYpLo_LqrOI4IF7Z406HZKgpp98QoGFQ8OBcry8,19942
flax/linen/initializers.py,sha256=CC4lY4uvAGisaxiiIAE2AHVPuN9q8EqQ2e9OZBDPi1o,2676
flax/linen/kw_only_dataclasses.py,sha256=_INxmh3LIPrTaBMfO44PdK8BqJkCeDCSR9vIiOn7G-8,9058
flax/linen/linear.py,sha256=7ngRoX-sNEmCorBiG7ROT8HIXB6CAuW17ZrxZtZRZJk,47275
flax/linen/module.py,sha256=NALuGq_pCwTi9dxDFa67huR-I3a718zGZxbeKqPccPI,117418
flax/linen/normalization.py,sha256=PdnmO9s9H6mQw9DhuvO1ER4cPWHnnZYUHTQEqjRCi_g,51190
flax/linen/partitioning.py,sha256=HRYexg6NWgm4xxL_dUqQbSPLZa2ozxNqfTZDUwj12Dw,19713
flax/linen/pooling.py,sha256=HNrjOv_ul5QEz27nXg69iSIzfRa42c2cbZh-q8Dciog,5567
flax/linen/recurrent.py,sha256=bcj8_f6G_Pz_4JUOR3p2B9oLCaxeEGHxUSR2dfrY2gQ,45837
flax/linen/spmd.py,sha256=ZUuiy5_98CQjuiaY7e_eRA4d0VCHS1X1V16k1X0NqGY,11555
flax/linen/stochastic.py,sha256=bvSVnu__BOvZkLaljmbXYtsNyVuc03zmatWUQ2SuSbg,3628
flax/linen/summary.py,sha256=MGXYpqY_7_CuZs8eJW5Ly59clnQLNXnV0xkCP79TC-A,26164
flax/linen/transforms.py,sha256=B1OD4qDGYpJlwEiKVXzcoxe7_wpYzm_50qdAkdIHcn4,79349
flax/linen/experimental/layers_with_named_axes.py,sha256=uR1mraQ9lAXM7LQaFX6dzV4nQa6maw3iGusCE5Uw8lQ,11231
flax/metrics/__init__.py,sha256=XCxb3apNmuctXSifbNcZMwlDQ-moSYF1SBxg40Y0UCY,581
flax/metrics/tensorboard.py,sha256=HUDQYfdhZXTCCxQJMpr9aUzrqexGZasEGvzu9ccy-ug,7792
flax/nnx/README.md,sha256=EQrb9TA1_UsdhBxRf8-AZZVcJDcM0CnonPUbyGu5dzA,3382
flax/nnx/__init__.py,sha256=p5-gXiyGMwq2WrfCY-lndFvULphKm2dphoGivQfp2XA,10467
flax/nnx/extract.py,sha256=6cLkFATZjCrM41T96JYdsgdXsKdjjhXc8q6J7A57WS0,13912
flax/nnx/filterlib.py,sha256=mehxnO6vqxBlpBCRlurI6Kn5FO5-hpAUbipZZNpmNmg,5397
flax/nnx/graph.py,sha256=EGuv16ewef0DLT0JaIwKzh7CcUnOcCJGXG8xKZUlwfg,91627
flax/nnx/helpers.py,sha256=-jszkhzAiSZsVMyDCrhkUbDKDAKvfrUvrbsA8KUeNJY,10281
flax/nnx/ids.py,sha256=mAFUaRQrB89E0fg1R2bzehpL09HOQSyceoRwF5RDC_E,1743
flax/nnx/module.py,sha256=1UsB8wcXiHKAHqx4eBnq1ly2Xg0kUX7-LHz93qfYsAg,25081
flax/nnx/proxy_caller.py,sha256=UqSWnOCB2anBRQgeEFZ9zAUOnPXSJCoBCp9PKoPpcdg,2361
flax/nnx/pytreelib.py,sha256=I0cppU5R37mQFqcmNTN7be50RDrT_-Crct5cjmhKq2I,32870
flax/nnx/reprlib.py,sha256=v3azacyufk4BJEcPqrt7L8i6dMgy7Kx2w4zS6MMVZC0,7255
flax/nnx/rnglib.py,sha256=HaCC3RPZ5BiQ-y2hm5sgJAsV31WIqLfTtMAkc_1gi9E,38364
flax/nnx/spmd.py,sha256=yfVLqhSGvnd2khgKrq6jCIanDdP2jdUZ8lhZxH7kzAM,5628
flax/nnx/statelib.py,sha256=EcT2uf99a0UdrFiAZq1Sbf1QefRhzW_W2cWdFst97Qg,24309
flax/nnx/summary.py,sha256=QCePrEAuaGcN_xuTQgLpWnCL8-ArGRycmMfLFVpz04Y,28672
flax/nnx/tracers.py,sha256=ZyNLsu1zyJHcfF1zasgEOesLEmVKjCFmeIuWV4slBiM,2273
flax/nnx/traversals.py,sha256=NiPdm7VS4Tqa36lOnmSm84abl-x-TibsKpkygL5AsyE,6238
flax/nnx/variablelib.py,sha256=LmECkjTkwwiDrU4-d0rvLhMc4YYhO1TVfyZ5kzMZtu4,75768
flax/nnx/visualization.py,sha256=344vI1OijvzAi3ZRosg1b3bdzEclu8rJE_pVDkELh4Q,5168
flax/nnx/bridge/__init__.py,sha256=vAaC4mCzx-XnH91vZeeHlZRjYpVA3PB0cSM2zzYl4YY,1392
flax/nnx/bridge/interop.py,sha256=4y6hNPvA1xsgVyISfF4zvJNHqwiVz-xJYicbgScVgUw,3641
flax/nnx/bridge/module.py,sha256=pvozbXCF0OWYgzLCen3QmIvm4QrDQlSfdjKL8PpDUFM,16762
flax/nnx/bridge/variables.py,sha256=lrnubvyoLTgf3mdEXjtYS1eGHLmHuVaNBkTYVkh-tQ4,6244
flax/nnx/bridge/wrappers.py,sha256=asQCgXFjxIJlbzHBxYiDMOUHhEXU0CNPTR_63hZpqH0,17617
flax/nnx/nn/__init__.py,sha256=ECrkYim1U12kMsc_1l0r-6drbITTHaO64JB5LZCuV7A,581
flax/nnx/nn/activations.py,sha256=s7bL18gCrdAxxRmhOYRK5qHTY1BuibUmBzZ-Oa9Ckyw,3649
flax/nnx/nn/attention.py,sha256=t2z80OD0s52k5RiknEkmp6G5MMxZU5HvNclnjNxGV8Y,34031
flax/nnx/nn/dtypes.py,sha256=Lpsj2tzufVerQ6HLOFhfJgCwKncHOQqLvw25UPRtVxQ,3293
flax/nnx/nn/initializers.py,sha256=0bfgv2mMU6iROZwzoyHOJbYpCm4--mjp700rJ5er1dU,2570
flax/nnx/nn/linear.py,sha256=BXldmsH7XEIq1X0qMTFJwoLyDK2aRVAQ1ISS_UuwbGM,50929
flax/nnx/nn/lora.py,sha256=1t2kz2oPAX3AjiJGQFZCqp-5vy7uOvN_f81hWsNA41g,7372
flax/nnx/nn/normalization.py,sha256=mGtLUxR5wdMpOsoJs0a3joUt4NxslK5xZGCo3AbHnPg,53524
flax/nnx/nn/recurrent.py,sha256=Dbh4sFdAO_l2Pw7nbQK-7Q0FxzMsFZPKkZMAlScFVho,37148
flax/nnx/nn/stochastic.py,sha256=f79ENHqRcojMctnnzAj7C-zyItUraLm5GiP0U0bED7A,5471
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=S6YwrzhoAPe5AW0aeDa7gBoGm2WH0rKMBgtVjjC403Y,15208
flax/nnx/training/optimizer.py,sha256=Dcr6Ip6YGU_uVYJ0QW0zOrWbnYV8IrKVUVd9G6qicpI,7796
flax/nnx/transforms/__init__.py,sha256=ECrkYim1U12kMsc_1l0r-6drbITTHaO64JB5LZCuV7A,581
flax/nnx/transforms/autodiff.py,sha256=GYJtl-Ka3za-gDbH04u4xVFNHg_l8qHQBWQYHoZ9-vs,30067
flax/nnx/transforms/compilation.py,sha256=8YB9unswBCRTqeqOcbkUpRRfUtzQHbAMlyLOK5KxUY4,49231
flax/nnx/transforms/general.py,sha256=FPe5uE9IYz8ZY8dRsKCUXxe93wsDckXWFh7bwPBiz7A,6938
flax/nnx/transforms/iteration.py,sha256=u7FhL_aI4m3fxM8nv2sDvr7NTXQq22NhbfmzFG4L5fI,52593
flax/nnx/transforms/transforms.py,sha256=3F1EeHI2dc3cAsaQ19GjsGwmE_QhqzmyORZ7o0dxuEI,11187
flax/oss/ .git-blame-ignore-revs,sha256=eJmLQj4P4rVEN8uRkflxwOhLFte29Y0WeGoI1uRNWfY,443
flax/testing/__init__.py,sha256=UarXpjywrTaGYJBYPf0joEzJawWavRs_f58YZJowfh0,647
flax/testing/benchmark.py,sha256=g_GkJcLn3MqSP_6aIhy7jQW65omQeeLyw-vA8-i-WDg,9307
flax/training/__init__.py,sha256=uJTipyYX83XazXl5bkf17EF4DjJE5s1YhCA3vS0vXqo,613
flax/training/checkpoints.py,sha256=xEGmtKlsHUnRulBEOiYexvDyFywFurkMaztsxegD-EM,43663
flax/training/common_utils.py,sha256=FtmDIp-5TFQtBnPqFDhr6ebCDcobNjuHIlgdQyDpoms,3987
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
flax-0.12.4.dist-info/licenses/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.12.4.dist-info/licenses/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
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=pwnUQ8_hOWQduJKk27pBy81eKNiAQThT1Qx51oCgscw,14979
flaxlib_src/src/flaxlib/__init__.py,sha256=2JE6AGH5She07uin4G443yt_R1B5MveMc0Lo7LBhP4c,810
flaxlib_src/src/flaxlib/flaxlib_cpp.pyi,sha256=-JdwRo3KQ3M2OflKNGQaeOd7VD-4Wrfg5Dx90tNLgjY,1905
flax-0.12.4.dist-info/METADATA,sha256=zI8uHk1cwLASIGEnkENc96ggfEa4sTiureAZwDgml7c,11739
flax-0.12.4.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
flax-0.12.4.dist-info/top_level.txt,sha256=ob6CnPYMaocT-XGMXv3h9nSaKwe-A1SaSn6Qeg2Xfec,17
flax-0.12.4.dist-info/RECORD,,
