flax/__init__.py,sha256=rdMmFz7H-NUM54KBz4vTh0Eb8IccznXWiit1J2BLHeA,1135
flax/configurations.py,sha256=BEFp6YUShXW0SqaX2eifbBMbGdKk6hoL7Bj0MyHB-0A,9254
flax/cursor.py,sha256=wAUA4Dqq6CFPH1L1SpztenCfrCZhbRW61JCFNJzeYTc,26386
flax/errors.py,sha256=Dz9kBdR5yPF6VaG7FARpe9i2Oafl7zdm232x8cfAVGY,32192
flax/ids.py,sha256=sbaPq1x8I1hGtZH4VVBufB9mKjfjvhUmwlOks4pN_O4,1752
flax/io.py,sha256=GYViY0BRwzN2i0T6TQ65eJQ8Br7Ye81udlruJXtkJ2Q,5309
flax/jax_utils.py,sha256=FfknUqVoFd8YoupZ4bI3asDVWT8GVhc4ZPCTx44xacs,12921
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=-T0UG5Pyg9MCLFqU-F_t21rS1K1yHeHpeBQsxkuxlpc,14327
flax/struct.py,sha256=tPaZBqS-I4fgFMgvspKupj2CkmWQguVw2F_CSBu2R4c,7801
flax/traceback_util.py,sha256=jUe3SrwCdlFM-rss-psgWVYL6px-Rd8dStxkhE8g81M,2085
flax/traverse_util.py,sha256=Q9rFcUQlhaGgDwVZJvBBL4vJKFL1hQ7hEzZ0zikUW8Y,13840
flax/typing.py,sha256=STRWCsElZflVVtafDm9pFA2cxv2ce3n2-EbjIyuhBeU,7943
flax/version.py,sha256=AU8Snhpw9BaqV_xVE95vyCivllXfu0mS-VICEJa3Ty0,651
flax/core/__init__.py,sha256=G2TuucsjMo9GsYkPcGrs_AxxKZ1osNbxU7De9CRfJuY,1436
flax/core/axes_scan.py,sha256=_0-i3H3l3FWQ86DYeiObabQ2ONBjHDyOIIL6maUJTRs,8609
flax/core/flax_functional_engine.ipynb,sha256=DGo3J5pSMS6BaDhQKUopSAt6NOQ8dp1lF0v8w63clMo,10482
flax/core/frozen_dict.py,sha256=d6LlT3Qwh7QqgvFOA1orCt3Hl06OJ6oWuBIUBQ0oRtQ,10301
flax/core/lift.py,sha256=YDRGX1Q4MzsvrUsyayazacrKJUNhgMWUqJc_fTprB9M,66896
flax/core/meta.py,sha256=nnZ2Ol8vDOhEMvB2IMFidjgk6NgLwN3_OEhTuLPOu7A,12638
flax/core/partial_eval.py,sha256=-wEOAjGjV9qbN9v9NCEFCMYrIiDHdRN35WfIRoXD7i4,2650
flax/core/scope.py,sha256=a_7Uf3QIMvU6jOgWcx-EL2FFJ9KLdCj64jFRuo2Vir4,37660
flax/core/spmd.py,sha256=YtZO7Nv71Y2WVR9mfStHiyRXCPaYJlpXG1QAI0vraqU,4979
flax/core/tracers.py,sha256=ImLRxsSCCyFXmgoBeB0xsBtIvQ9fSc-BDNbqxsX9_3o,1396
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=QZWHYcVprsqGTY6-xOV0Jyf_Amf7gSI2M6Oa2UrVVeA,47271
flax/linen/module.py,sha256=kPoga779lmeEDdMrfMD0slqJljcuc7DcQVHT5D46he4,117414
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=0VG8N-xw_mfBMCQjJeqvOrrpz-4yb8Kb51_Lh-FFY50,79357
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=XxOIt3FzmEAo6QVwqzHS6VRC8ss7KLUhMGSjGGrWPyk,11668
flax/nnx/compat.py,sha256=Y0_wLgu5x_aYnQk7gKKIXGNvemyJq4ahXnRTLADMvSg,5183
flax/nnx/deprecations.py,sha256=yiGlVgaM00kiSwlbROrpPCOYMfJ5EcFJyNrEk-OxaXQ,2065
flax/nnx/extract.py,sha256=UxIEIut2cQ3e6OIH6UgoJFsaZb1WWuvix0oJ8oLEO5w,34919
flax/nnx/filterlib.py,sha256=mehxnO6vqxBlpBCRlurI6Kn5FO5-hpAUbipZZNpmNmg,5397
flax/nnx/graph.py,sha256=8aPEfqTbR7fZl_n5jlbHN4wvitaqo9H37-vovr8dvBU,776
flax/nnx/graphlib.py,sha256=Vtfi9Z9lbgkDi1kKeN9BmrK1HiyURHD_K8NDTpK9lew,118156
flax/nnx/helpers.py,sha256=5T0dU0PdQBmvGkMTJ_Rm0nspFql1XcgguUEfQcx_aFI,10445
flax/nnx/ids.py,sha256=mAFUaRQrB89E0fg1R2bzehpL09HOQSyceoRwF5RDC_E,1743
flax/nnx/module.py,sha256=fY9OxZiKhzUq6M2s3ubl-zjxURKJNwItunX4iRhtB6c,36022
flax/nnx/proxy_caller.py,sha256=UqSWnOCB2anBRQgeEFZ9zAUOnPXSJCoBCp9PKoPpcdg,2361
flax/nnx/pytreelib.py,sha256=h3Vg8QIl0wCzaZ_ieyXKPACm_48EmIjAf_1buGpYUSw,34272
flax/nnx/reprlib.py,sha256=v3azacyufk4BJEcPqrt7L8i6dMgy7Kx2w4zS6MMVZC0,7255
flax/nnx/rnglib.py,sha256=DP2hEJkvgrWzKPxE2k_0HX6BvIkxXmFHvXqtZrCL-Hs,54731
flax/nnx/spmd.py,sha256=yHELzp0HvcpR5_jnru85VLQJp6NJG6urkCt4DH72q98,8097
flax/nnx/statelib.py,sha256=v6FwjRvbxUS9xBsJ-VMkdJOUobNHXRGPYxyM1tkSuCs,24452
flax/nnx/summary.py,sha256=8awLcelKr8-O7wf698Q2BlQIu-w6JPLXpvgI2KmTMXk,28693
flax/nnx/tracers.py,sha256=RlkIrYP01fXWuJc2Wj0AJake3I5dgUlhoyXVFl7YaMQ,2186
flax/nnx/traversals.py,sha256=NiPdm7VS4Tqa36lOnmSm84abl-x-TibsKpkygL5AsyE,6238
flax/nnx/variablelib.py,sha256=PZXHy77zbPOIM1cx4_yjCK3rs6Lodfp2czLI8ugUg2o,74868
flax/nnx/visualization.py,sha256=344vI1OijvzAi3ZRosg1b3bdzEclu8rJE_pVDkELh4Q,5168
flax/nnx/bridge/__init__.py,sha256=vAaC4mCzx-XnH91vZeeHlZRjYpVA3PB0cSM2zzYl4YY,1392
flax/nnx/bridge/interop.py,sha256=nUhU6M0rzUBEAXgPCBqzIFG0a4u6NBQK5TGU1dMWMKk,3674
flax/nnx/bridge/module.py,sha256=6R-GkbYpFgDYEOf0s_OrYairaQMKG8fU01w6m7eyj4E,16157
flax/nnx/bridge/variables.py,sha256=lrnubvyoLTgf3mdEXjtYS1eGHLmHuVaNBkTYVkh-tQ4,6244
flax/nnx/bridge/wrappers.py,sha256=M9Vfjco979VGo3fGJ0OlYDlZak2X4q97-y_Cg3cjM9E,17793
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=pZ3O94cc8ZeGBNFypApuD-rre7uuVSA0txKomiZtthw,35748
flax/nnx/nn/dtypes.py,sha256=5JmOPCy_n7HfBl7nkDix4Rpy8Hg0FWWQDIfRwUokf3Y,3444
flax/nnx/nn/initializers.py,sha256=0bfgv2mMU6iROZwzoyHOJbYpCm4--mjp700rJ5er1dU,2570
flax/nnx/nn/linear.py,sha256=BjshwNLHfUcjsbTWG1kh8rJQf2byTKmTg0yEHFjJNSE,50925
flax/nnx/nn/lora.py,sha256=OJEJ2ehxkLMZJXVsv1MdNhKxHf3KZaxysx2YLPI31CY,7400
flax/nnx/nn/normalization.py,sha256=lU4slgXbXb_WngqPpbwD3T2Ph9QeMoxyPBfx39sO6wU,53981
flax/nnx/nn/recurrent.py,sha256=KH_PyT2qsYG3QHBRXeIlyl-W34IgS7s9W15syOfJl3Y,37634
flax/nnx/nn/stochastic.py,sha256=Js8qit9ADFL8zi5VEhQykctZRGUiZ8I99otRjCMw55c,5906
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/ema.py,sha256=1qSd3AcdBMmWpf6qsm2UyLGL4mF-ao-YXqM2x3_RUKA,6361
flax/nnx/training/metrics.py,sha256=J4sUN_Hgayfbbx7GSKGzv6s51AukAHzTptj5yHgenak,16522
flax/nnx/training/optimizer.py,sha256=bFSMfZrU-VWSsO85pzXdCYVMVZIzRn00hwt92zOOQsY,8343
flax/nnx/transforms/__init__.py,sha256=ECrkYim1U12kMsc_1l0r-6drbITTHaO64JB5LZCuV7A,581
flax/nnx/transforms/autodiff.py,sha256=FANCWlHCK0THbfCa11gkSJhl-dIO16woYsvKSYFxtDQ,56814
flax/nnx/transforms/compilation.py,sha256=9EXOwwP1TYAdMSdwbSWdHv260QoxuyT2cGO9G6cipVg,68904
flax/nnx/transforms/general.py,sha256=K0cw2xojEUgm906Q4xqnUx4TnRr029lO5PREwIAy1J4,6944
flax/nnx/transforms/iteration.py,sha256=E4KIUf7Gxg53rx2V-JHnWljSy2QpzAAEM8hxXdWuq9A,71696
flax/nnx/transforms/transforms.py,sha256=gdDRmA5KuWo6fR4cLfXV-ta9chV0_mt-jvSfuED-wA0,22441
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=9hr9AdhO3PpFAiVCb8PjLhVc_EAhI3oArN7RxSaIlmQ,43725
flax/training/common_utils.py,sha256=TzMRUtEXJsV94M8THaObkx686iHs7Jp-UelaljoGowQ,3718
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=lLvE5OA-rYo0mg8pNLakc53XZmJlXaXrCxXdXvjuNpg,4842
flax-0.12.8.dist-info/licenses/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.12.8.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.8.dist-info/METADATA,sha256=KDeLj-oKVJKRYFU89JYcgB7DJ538aFEwSOOfbpcLX7E,11760
flax-0.12.8.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
flax-0.12.8.dist-info/top_level.txt,sha256=ob6CnPYMaocT-XGMXv3h9nSaKwe-A1SaSn6Qeg2Xfec,17
flax-0.12.8.dist-info/RECORD,,
