flax-0.12.7.dist-info/METADATA,sha256=e6J8piYUL5H5EXVRgnpCPysCZxN6qb2DSfaQGlV-Fpo,11661
flax-0.12.7.dist-info/RECORD,,
flax-0.12.7.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
flax-0.12.7.dist-info/licenses/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.12.7.dist-info/licenses/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.12.7.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax/__init__.py,sha256=rdMmFz7H-NUM54KBz4vTh0Eb8IccznXWiit1J2BLHeA,1135
flax/configurations.py,sha256=q7q24KTSsK9O08qLctHhTaFF6tHUf5gtXj5c6-2G5zo,9076
flax/core/__init__.py,sha256=G2TuucsjMo9GsYkPcGrs_AxxKZ1osNbxU7De9CRfJuY,1436
flax/core/axes_scan.py,sha256=_0-i3H3l3FWQ86DYeiObabQ2ONBjHDyOIIL6maUJTRs,8609
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/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/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/cursor.py,sha256=wAUA4Dqq6CFPH1L1SpztenCfrCZhbRW61JCFNJzeYTc,26386
flax/errors.py,sha256=Dz9kBdR5yPF6VaG7FARpe9i2Oafl7zdm232x8cfAVGY,32192
flax/experimental/__init__.py,sha256=hkK7vneGSpZel3JlyAuXLe-tggix1dk6NsSYwhQXOdE,582
flax/experimental/nnx.py,sha256=YMLSuRAPHEJLdILaiU8_k7rgZ8KKadeYHmoWCq8ux3I,731
flax/ids.py,sha256=sbaPq1x8I1hGtZH4VVBufB9mKjfjvhUmwlOks4pN_O4,1752
flax/io.py,sha256=GYViY0BRwzN2i0T6TQ65eJQ8Br7Ye81udlruJXtkJ2Q,5309
flax/jax_utils.py,sha256=FfknUqVoFd8YoupZ4bI3asDVWT8GVhc4ZPCTx44xacs,12921
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/experimental/layers_with_named_axes.py,sha256=uR1mraQ9lAXM7LQaFX6dzV4nQa6maw3iGusCE5Uw8lQ,11231
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=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=0VG8N-xw_mfBMCQjJeqvOrrpz-4yb8Kb51_Lh-FFY50,79357
flax/metrics/__init__.py,sha256=XCxb3apNmuctXSifbNcZMwlDQ-moSYF1SBxg40Y0UCY,581
flax/metrics/tensorboard.py,sha256=HUDQYfdhZXTCCxQJMpr9aUzrqexGZasEGvzu9ccy-ug,7792
flax/nnx/__init__.py,sha256=rXFtZBznBs19h5iWj7HTgAibndkfJp1YuUqJKx1KAL0,11586
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/compat.py,sha256=Y0_wLgu5x_aYnQk7gKKIXGNvemyJq4ahXnRTLADMvSg,5183
flax/nnx/deprecations.py,sha256=yiGlVgaM00kiSwlbROrpPCOYMfJ5EcFJyNrEk-OxaXQ,2065
flax/nnx/extract.py,sha256=sWJ5texeE77Ov7EWZpgI5nugVUQJ2jIfdZ11yBIZC-A,31457
flax/nnx/filterlib.py,sha256=mehxnO6vqxBlpBCRlurI6Kn5FO5-hpAUbipZZNpmNmg,5397
flax/nnx/graph.py,sha256=8aPEfqTbR7fZl_n5jlbHN4wvitaqo9H37-vovr8dvBU,776
flax/nnx/graphlib.py,sha256=X0ynVCtcYJClgMjam1AcPqsP-hVa-40Ae8KzPOpYOg0,116846
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/nn/__init__.py,sha256=ECrkYim1U12kMsc_1l0r-6drbITTHaO64JB5LZCuV7A,581
flax/nnx/nn/activations.py,sha256=s7bL18gCrdAxxRmhOYRK5qHTY1BuibUmBzZ-Oa9Ckyw,3649
flax/nnx/nn/attention.py,sha256=z300WD_wjmqrFGtiiqMb1NxNI6tR8_jVi1FXrFlBxfo,34740
flax/nnx/nn/dtypes.py,sha256=Lpsj2tzufVerQ6HLOFhfJgCwKncHOQqLvw25UPRtVxQ,3293
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/proxy_caller.py,sha256=UqSWnOCB2anBRQgeEFZ9zAUOnPXSJCoBCp9PKoPpcdg,2361
flax/nnx/pytreelib.py,sha256=vxOyzMgJxKnp5V3GXEMTm5tNj3IWdYcOh6w5ltDkkNA,34120
flax/nnx/reprlib.py,sha256=v3azacyufk4BJEcPqrt7L8i6dMgy7Kx2w4zS6MMVZC0,7255
flax/nnx/rnglib.py,sha256=3Od0oCP2Z3npQpBx5kj95LdzANo_f0WPWDqDIuRUK6A,53306
flax/nnx/spmd.py,sha256=Ki7GgpkhEPpN-1AwTSS_iYLnt9yd97SCMcdhmw_8WXQ,8108
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/training/__init__.py,sha256=hkK7vneGSpZel3JlyAuXLe-tggix1dk6NsSYwhQXOdE,582
flax/nnx/training/metrics.py,sha256=DoE8l1vZKECJTzzC9DvIGty57WFu8plXAAM3KGXdXgQ,16548
flax/nnx/training/optimizer.py,sha256=QJL472xwEizPtx6xQFwusFgH-OEc1epRpV5618Vue8s,7910
flax/nnx/transforms/__init__.py,sha256=ECrkYim1U12kMsc_1l0r-6drbITTHaO64JB5LZCuV7A,581
flax/nnx/transforms/autodiff.py,sha256=Rsa3mIeAPEuokD6L-4JswYlAi_neB3uTK7--08cNMDs,56137
flax/nnx/transforms/compilation.py,sha256=1iuWT8VyOj3ZCtIfDmmSQpHuGFNZj7SQvILDw_QJNoA,60647
flax/nnx/transforms/general.py,sha256=K0cw2xojEUgm906Q4xqnUx4TnRr029lO5PREwIAy1J4,6944
flax/nnx/transforms/iteration.py,sha256=8OifP5JXzFhdCN6jzMTnzOcOkE-7QnrWNJ8iGEMZNv4,72333
flax/nnx/transforms/transforms.py,sha256=tJg2qwUh9Doe5h8T766LNAMIvje9C2nJhrdQBqmEKcg,22255
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/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=EKLEl2OD0mrLNc7iriYZuLwNTuuMoeMMOy1MtpP8oVI,14305
flax/struct.py,sha256=tPaZBqS-I4fgFMgvspKupj2CkmWQguVw2F_CSBu2R4c,7801
flax/testing/__init__.py,sha256=UarXpjywrTaGYJBYPf0joEzJawWavRs_f58YZJowfh0,647
flax/testing/benchmark.py,sha256=g_GkJcLn3MqSP_6aIhy7jQW65omQeeLyw-vA8-i-WDg,9307
flax/traceback_util.py,sha256=77RUsCJNAGRVOG3aCQIQQkNenl-1ztZ1cmNS59J0Rrg,2075
flax/training/__init__.py,sha256=uJTipyYX83XazXl5bkf17EF4DjJE5s1YhCA3vS0vXqo,613
flax/training/checkpoints.py,sha256=xEGmtKlsHUnRulBEOiYexvDyFywFurkMaztsxegD-EM,43663
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/traverse_util.py,sha256=Q9rFcUQlhaGgDwVZJvBBL4vJKFL1hQ7hEzZ0zikUW8Y,13840
flax/typing.py,sha256=STRWCsElZflVVtafDm9pFA2cxv2ce3n2-EbjIyuhBeU,7943
flax/version.py,sha256=17MyOqaBsvrZTX4ubImLS-tVg5IK1W6BJ3bW_XTj7g4,651
