flax/__init__.py,sha256=rdMmFz7H-NUM54KBz4vTh0Eb8IccznXWiit1J2BLHeA,1135
flax/configurations.py,sha256=5bK1gYUDtADBEb656zqXPxgskFIwJ44A1RL6LwAdxGQ,8256
flax/cursor.py,sha256=76hSxZHDy5x0T5og8Sz5MOoAcDVS7pwbUORn8DxEzCI,26394
flax/errors.py,sha256=kMBWM9ETNJ5V5mSn1W7cvsJ9CbOPxgV4pxS31owLWW8,31682
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=4paUjwwyGtPbZc7ndEG6alJg6RB07NjEvHul4CRU73I,7769
flax/traceback_util.py,sha256=77RUsCJNAGRVOG3aCQIQQkNenl-1ztZ1cmNS59J0Rrg,2075
flax/traverse_util.py,sha256=mTTx6l1r69-o36n_Qx4eenxDWoNtxjkT2Bm_j1fGrIc,13773
flax/typing.py,sha256=mC6iQnoBEJpelNcyL-IqIm-6jNTvUQC-wCDlgdMOJOk,5283
flax/version.py,sha256=EA9VPAMrjcFYfTGioJT5dncvCEe6PcZXdHqghuwaJ3I,651
flax/core/__init__.py,sha256=G2TuucsjMo9GsYkPcGrs_AxxKZ1osNbxU7De9CRfJuY,1436
flax/core/axes_scan.py,sha256=S35lGPoHJazISPgHm-8rBKb2gZnPoEMJjOnj0dS4WWM,8592
flax/core/frozen_dict.py,sha256=d6LlT3Qwh7QqgvFOA1orCt3Hl06OJ6oWuBIUBQ0oRtQ,10301
flax/core/lift.py,sha256=uSGg2yXHEErvJ7P0p15NenFEYhQ4QoKtkr8Hp7eyaAg,65703
flax/core/meta.py,sha256=j1Inho4A_AHegfOyC0H9S2ZQ8H18oJcIJU3kC-oDJVU,12748
flax/core/partial_eval.py,sha256=-wEOAjGjV9qbN9v9NCEFCMYrIiDHdRN35WfIRoXD7i4,2650
flax/core/scope.py,sha256=XSHNlAQvjxxeX0KHfP-DRZCM5dqXmpZ8UseCM26jgbI,37301
flax/core/spmd.py,sha256=wOWx2qdP7q053r9ANMYNaJsM7M4UCswNbmReUtWEwQk,3497
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/__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=eVHHSHiIN1aut4Y1bdiYXkVh2dYAkqp0zo-QuClF4yA,19947
flax/linen/initializers.py,sha256=CC4lY4uvAGisaxiiIAE2AHVPuN9q8EqQ2e9OZBDPi1o,2676
flax/linen/kw_only_dataclasses.py,sha256=U4-mGzgAhz902A1SL29J8Hdr33smxl8NUB9BwOz-4GU,8828
flax/linen/linear.py,sha256=DpBllX1Xn_1oPAS0r49mA9kFmk95_kAxLxPT5_EIijU,47036
flax/linen/module.py,sha256=QZg5bKRj8efHrLH5jiOFodyEmJE3UH6pwcKEDbGR3Qc,117321
flax/linen/normalization.py,sha256=-JkxYpHotIcSFT3ixVgU1tL60PDPUV6WW00Qcrl9lgY,50802
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=JF0tmYMg0gciZ36HqhMdhcThhXUjie85BaSrtnDmu6w,11018
flax/linen/stochastic.py,sha256=bvSVnu__BOvZkLaljmbXYtsNyVuc03zmatWUQ2SuSbg,3628
flax/linen/summary.py,sha256=MGXYpqY_7_CuZs8eJW5Ly59clnQLNXnV0xkCP79TC-A,26164
flax/linen/transforms.py,sha256=b4pzZWGLYJG7Ajq4lJO-uDFG4z-zr9a_yK71ZENi0oI,79350
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/__init__.py,sha256=kDlunlkeQlJcNQRZZc7-yO30DxrFziaTpQMrJGD8ad4,9714
flax/nnx/extract.py,sha256=TOvlINfTE12nZ5tFkTVbsyW9L1iS3aNkyn5AwCECP3s,9959
flax/nnx/filterlib.py,sha256=2Iw6v4xpwAHhE0j5EX7R_6kZQCNCNJXrrJij3U2dOLY,5269
flax/nnx/graph.py,sha256=4GsKLCBjdcNN-Myqjb4rlQRPLCMGlR2gXJy1jsPsQMU,94702
flax/nnx/helpers.py,sha256=OVGCna6V-oeOMPh2AFZKrvhRqivDEl0ttOSs2I6IMes,4157
flax/nnx/ids.py,sha256=mAFUaRQrB89E0fg1R2bzehpL09HOQSyceoRwF5RDC_E,1743
flax/nnx/module.py,sha256=dp1j5im-LrqvJ-dncBa7CyYV-VylcpCPRGrV4d_-kMo,17974
flax/nnx/proxy_caller.py,sha256=UqSWnOCB2anBRQgeEFZ9zAUOnPXSJCoBCp9PKoPpcdg,2361
flax/nnx/pytreelib.py,sha256=mkGcL0DQxldqf2vXvvaYcqPV4t-u1Cjz26UgG-KoqHM,18430
flax/nnx/reprlib.py,sha256=neWTdzjxbmgaaHJ52uuDF0T4Pm4aEbYB2wMC7ONg-O0,7245
flax/nnx/rnglib.py,sha256=aHBg6PCyBqrgAr4vo8B2ZTXwL_SZwl0bz5OHqThhIEg,35369
flax/nnx/spmd.py,sha256=yjr7NrHHgTI7WkmXI7nU2Fsq68QdoV9oZVheDCIRv4w,5730
flax/nnx/statelib.py,sha256=XWOWOqJmg3Cos638YF-e9CDrJ-5buq0uis9ZrrYxk6k,21987
flax/nnx/summary.py,sha256=zECqLZCp-ZLv_mvs2ABryScT39lLxexafkGvn0y_7_s,23936
flax/nnx/tracers.py,sha256=ZyNLsu1zyJHcfF1zasgEOesLEmVKjCFmeIuWV4slBiM,2273
flax/nnx/traversals.py,sha256=NiPdm7VS4Tqa36lOnmSm84abl-x-TibsKpkygL5AsyE,6238
flax/nnx/variablelib.py,sha256=dCienWdM0POThTNmKN-9g69daf7uQdZmBg42Sdkb07Y,35466
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=I73mVmQZ1dxV6d5d9KA93UNgfvJAvLAeQFboau9Qegg,16746
flax/nnx/bridge/variables.py,sha256=gdzzR2J02ZXPuqQGBI20mFtTcuu0BT6dcSWqqz9WNyc,6247
flax/nnx/bridge/wrappers.py,sha256=WsFTJkznuLdDqsiRJMMBWXhpI7ydv4wCgby2wr9Eegw,13241
flax/nnx/nn/__init__.py,sha256=ECrkYim1U12kMsc_1l0r-6drbITTHaO64JB5LZCuV7A,581
flax/nnx/nn/activations.py,sha256=uQdzUr1hzSxgaA-Quvqyc1m7rWufp7fbpwReByYxXtI,1245
flax/nnx/nn/attention.py,sha256=xMmejQDSBh24yLM9AGXxoTVGzVEVbw5z2j0EgqMlqcc,26866
flax/nnx/nn/dtypes.py,sha256=Lpsj2tzufVerQ6HLOFhfJgCwKncHOQqLvw25UPRtVxQ,3293
flax/nnx/nn/initializers.py,sha256=0bfgv2mMU6iROZwzoyHOJbYpCm4--mjp700rJ5er1dU,2570
flax/nnx/nn/linear.py,sha256=ZXatPhai4TM2Mxld72JZDkeiCx3g7tXCHKqtbqDm-XU,43274
flax/nnx/nn/lora.py,sha256=C1YdraciSEcLFP0VS5d1K7SeSRRdwSRijXC8fHNSU0U,6027
flax/nnx/nn/normalization.py,sha256=WRfZh2zfbAr0-Pg7SGIGgms_3A9CWGuibgG3Ej-fsqU,29225
flax/nnx/nn/recurrent.py,sha256=clFHMyCk2aBCWkYLyzyZr1fOp9AXEMRePxnQuUxEKeI,31617
flax/nnx/nn/stochastic.py,sha256=PLRtT2fXV6y1j43TnRSB8RQJB0cJw5Lqe-2HtHtyTdk,5077
flax/nnx/training/__init__.py,sha256=hkK7vneGSpZel3JlyAuXLe-tggix1dk6NsSYwhQXOdE,582
flax/nnx/training/metrics.py,sha256=5L1647I8Ai9wHZtML1hkes9b8hkIZLfpHDZJOZLZdhU,15266
flax/nnx/training/optimizer.py,sha256=e2Jgdb7nrFPaCVNxlHSYsbFGhkpyh1lomex7VO0JGXQ,7831
flax/nnx/transforms/__init__.py,sha256=ECrkYim1U12kMsc_1l0r-6drbITTHaO64JB5LZCuV7A,581
flax/nnx/transforms/autodiff.py,sha256=jAA_R1G5mw1S-sx4Ux5G-4Tn0n4QS847F6DP0NomR_Y,29138
flax/nnx/transforms/compilation.py,sha256=0-JG5LZKOXFH0ye4J2AA3TIqlzUagemnk0QVbgmn6Dc,36049
flax/nnx/transforms/general.py,sha256=F7JQCB_WxN0vv0h2vzbPnzmBE-nUjArS4dOfR6USXso,6933
flax/nnx/transforms/iteration.py,sha256=EPYr9Kv9plFZQjqyLeRxTd5u0nsX2DBhhgTtqjdQ2pQ,49099
flax/nnx/transforms/transforms.py,sha256=T-Cv6bti41BauChm6QXIxMhE8AS47Es1unZ6SI6p2oA,7168
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=_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
flax-0.11.2.dist-info/licenses/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.11.2.dist-info/licenses/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.11.2.dist-info/METADATA,sha256=ebmnTj5sA4yQQ5hbx1bjnGn-uDNtzlyLepk7WAIuEk4,11328
flax-0.11.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
flax-0.11.2.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.11.2.dist-info/RECORD,,
