flax/__init__.py,sha256=rdMmFz7H-NUM54KBz4vTh0Eb8IccznXWiit1J2BLHeA,1135
flax/configurations.py,sha256=ujrTueyd4ZDZ27M-KZ2zczVHc6sHzZxKN_URMMsZb_Y,5990
flax/cursor.py,sha256=mhIa8Czih_kUHfIV5Aq6YbjqutYzyPFi40oolHGmIlo,26339
flax/errors.py,sha256=7vnunxbS-Lhc7N5xy0IsNZ-RuDAQGgpF0ENBtFUzVY8,30713
flax/ids.py,sha256=sbaPq1x8I1hGtZH4VVBufB9mKjfjvhUmwlOks4pN_O4,1752
flax/io.py,sha256=GYViY0BRwzN2i0T6TQ65eJQ8Br7Ye81udlruJXtkJ2Q,5309
flax/jax_utils.py,sha256=FTtXCqYKRoxGTAeLLIYMGn0JFNAph6_LiTl5_cLpL0c,11534
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=LBi2n4uCXHpUQ9HJsiIjnlHu_Ae8y1S1QldiwuOtymM,14315
flax/struct.py,sha256=H7pFpf0GEFKdJnaITdunPzoFY7apUIMGF4oE5zDl1oI,7886
flax/traceback_util.py,sha256=VSBwID-DMPnPfXlESEkNlqZIVj009mF0kkdlvho8qXE,1990
flax/traverse_util.py,sha256=UGSxF9PK8Io2NXyJsrFy8ahYLsxXyfHeKVLMWQi02eI,13746
flax/typing.py,sha256=d_YGzVgJAjVGLWy1fPFVpO7h3YE_U8bjLQdSX8L81-g,2560
flax/version.py,sha256=KUD_t1Gd46u5X5Y4_6747nZarjj7iTFY2bk3DlwDFB8,650
flax/core/__init__.py,sha256=9gthSYfQPBSZvkUNAF6Rl6TzkUJqrEMcL1h4HqPQPWA,1467
flax/core/axes_scan.py,sha256=dDT7xT-3fRY75tit1SXFFudMsKWcDnyfrUKk699YkVw,5359
flax/core/frozen_dict.py,sha256=d8_mpjCi0p_LS6EHwx6-ZodLKMALvqSQIT2EhoTTYaA,9916
flax/core/lift.py,sha256=Htf6imISEqOPDcIwNEiRtushgtT4scjQ0n83B3EeoR8,60884
flax/core/meta.py,sha256=luEwggP23uh-nUAoYPUB7JyjNM6zsC32I_59fVkNZmE,11661
flax/core/partial_eval.py,sha256=hj_llMsUSSqH1eF6bUdnhWsWFpLlApOPj6Y9d7VbnOg,2546
flax/core/scope.py,sha256=5qDkt6X63Haib4JAiqr7HU_GOLkrqGhoiCKP11kbnVI,37240
flax/core/tracers.py,sha256=wzi47_1VH5SbozYVfnvtUfntbrOe2JDlKEjBM0wOwkg,1054
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=jlaJ0m42bjslMZQRK_0l2yYV8x1bY23pQGoxLqUziwg,17932
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/__init__.py,sha256=GEvWAPOKEJ9IbdnnNdH2sNUTkp53wUUpSEEUH1iZiN4,5172
flax/experimental/nnx/examples/lm1b/input_pipeline.py,sha256=-dqS1rhL0LPXHWNdV7_kOdDjF4yUzBYfOpKtlleOMjk,12344
flax/experimental/nnx/examples/lm1b/input_pipeline_test.py,sha256=xbsaCa9vNPfXLQV2s5zXh2wzELIZ4CQqzv4-TopAY7s,3293
flax/experimental/nnx/examples/lm1b/main.py,sha256=k28k3dKrQQVeQHYoJ8Yi-XyiJDNGz9r10uDTUaCq3Es,2143
flax/experimental/nnx/examples/lm1b/models.py,sha256=lxAFyTQttwyGCADgZ2gcLGgEKIGz68PxoeEXfuzHtoo,14978
flax/experimental/nnx/examples/lm1b/models_test.py,sha256=mFnWrevaBdDSNoxyzA5WF8N8GIhJrFXGa-rTB8jZA3E,9490
flax/experimental/nnx/examples/lm1b/temperature_sampler.py,sha256=tyoiOf4dwoLqkau4tOp7hilU2ocmafSDPYhFeo7oFX4,4799
flax/experimental/nnx/examples/lm1b/temperature_sampler_test.py,sha256=ATe15vuNrU51PlJVdablJ60Xn4hkOptUtrCgye5PaR0,1448
flax/experimental/nnx/examples/lm1b/tokenizer.py,sha256=JoQD9UhDe_ff7UnoAfrbvbL4Z-FDj1ZFRgwr3_x5_r4,5263
flax/experimental/nnx/examples/lm1b/train.py,sha256=QJVkyyquXhpeosdDeq37Kvz6xuQj4DZcrJUfO4l3Iu4,20453
flax/experimental/nnx/examples/lm1b/train_test.py,sha256=huVCYOMfppbJWNhTpnJaTnrtQWZ7sjjDHEz-KrWc3Zc,2026
flax/experimental/nnx/examples/lm1b/utils.py,sha256=lBPJXXdjjVDx9Cb6rL9uB5sEMirRCE4VHxdvM32kS3Q,5015
flax/experimental/nnx/examples/lm1b/configs/default.py,sha256=rp8zsTa4a4WK_29M57cletqNk6eDsT6xYmCuML24zJ0,5096
flax/experimental/nnx/examples/toy_examples/01_functional_api.py,sha256=SvPz5xl5wKNfwdjR1AncpyRzUaH3HS6p_dpntaTllzY,2790
flax/experimental/nnx/examples/toy_examples/02_lifted_transforms.py,sha256=pAp77ByfxIQrt1_TF_JyC7nCDqBEqtDlPMCZz7QZUAU,2547
flax/experimental/nnx/examples/toy_examples/03_train_state.py,sha256=xhb8eYz8uVqv9QeI9nXFIhfr2KIFeYuDDH53SjJgXYY,2860
flax/experimental/nnx/examples/toy_examples/05_vae.py,sha256=RW0UFvKAB1k6HHVZSBSmU0OMmBb2JbiOiCWNXmyoUhA,5704
flax/experimental/nnx/examples/toy_examples/06_scan_over_layers.py,sha256=ihZiMHKJ588VzdmX1Dw6rmzquUOAP7gcAQ4zlKr04tE,2782
flax/experimental/nnx/examples/toy_examples/07_transformer.py,sha256=fpI8mj5HgCXsFTfd4trQk_D-tUvRpstBGVZfQEJHAcA,12131
flax/experimental/nnx/examples/toy_examples/08_save_load_checkpoints.py,sha256=HxczamIuiXm4cWrE-wn4SrOeEe49TZy5xsI8w8cKYk4,1936
flax/experimental/nnx/examples/toy_examples/09_parameter_surgery.py,sha256=69coIqTsRuEz4yJ7Tm_t9dlXjMtOa4lBMvBkQg4xwBg,1940
flax/experimental/nnx/examples/toy_examples/10_quantization.py,sha256=P9Qyhayhua4z0gBLeIizsUT_yuNBTXU8b-a-Pp992OA,10896
flax/experimental/nnx/ideas/shape_inference.py,sha256=Q6hdqCJLDJBUlVwOgTaPb1_24NXhFT7veOi-YldLsMY,5520
flax/experimental/nnx/nnx/__init__.py,sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg,1163
flax/experimental/nnx/nnx/compatibility.py,sha256=lZXxDaA6KzHigE1-iPRsqWrXugNUZsioBKmDhk9oiX8,3176
flax/experimental/nnx/nnx/dataclasses.py,sha256=mMynSlNzY_-FBmSp2W4Vx0kvblwuA-3QGPRBbdTIh1U,4326
flax/experimental/nnx/nnx/errors.py,sha256=5FCT85_hNWEiikufgReDuwN_uC_4HxZj-VMHS7mIcOY,626
flax/experimental/nnx/nnx/filterlib.py,sha256=h-Hm529II5ZyHEK7kkvghvMg8aEc8lAfPC61z1H1oK8,2631
flax/experimental/nnx/nnx/flaglib.py,sha256=1zJdFDj_ItpDjNanoSRSB7AMaOd7fHsJ33-v2IdZc84,2109
flax/experimental/nnx/nnx/graph_utils.py,sha256=exYJLLywO0xB87U2MVVnOSetRb38Av54OJeuisVJSlc,24754
flax/experimental/nnx/nnx/helpers.py,sha256=VVoyXj9LZjMMsa0qUNjXjRlztOdlho5E6KDtoGVG07o,5598
flax/experimental/nnx/nnx/ids.py,sha256=vuURWDY8w7XWiRpUz3njkiVEciJZY87mrcmSTc07mjY,2326
flax/experimental/nnx/nnx/module.py,sha256=Ui4og0aPcgy-YtszqQpNfVBx_wutt24XQsrxK6uz7bs,16103
flax/experimental/nnx/nnx/proxy_caller.py,sha256=cgiCr36VJSsMeSsamLJDk0ry2MTCzFznBgjEZVFmaXc,2562
flax/experimental/nnx/nnx/pytreelib.py,sha256=PZwzURiRr4cLk3kD8Z__VLXcobrhVVkmg3r2la1mwVo,9099
flax/experimental/nnx/nnx/reprlib.py,sha256=APQG58t4WPgJJsasH5SOvTPt4jChKkZo6ChWIS3PzM8,2781
flax/experimental/nnx/nnx/rnglib.py,sha256=XS1-pRX9cL9OeI2nsxkvcz70XGgVITBnBkKJsit8RYU,6920
flax/experimental/nnx/nnx/spmd.py,sha256=sje2_Cx0CW_VGaWD6xKk5O12aPzKAda1xsCbEEViSKc,6395
flax/experimental/nnx/nnx/state.py,sha256=-hWpu1WEo4Guum0X4qZwNUxRBNbyAW-frDOuv37mboQ,9081
flax/experimental/nnx/nnx/tracers.py,sha256=_OIzYmJf1KJn8-PibjNkk_qnHLMWkfdYlRRvcuecBwQ,2866
flax/experimental/nnx/nnx/transforms.py,sha256=wcwP-tB4kvc-J_8xzeGNAo7XNskGSRY3LkImG-kKe7M,44496
flax/experimental/nnx/nnx/variables.py,sha256=NtVKY3SzEUjuioXD9PgKVX38OOk__-pfsYKmMDIFKzc,14987
flax/experimental/nnx/nnx/nn/__init__.py,sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg,1163
flax/experimental/nnx/nnx/nn/activations.py,sha256=Ub94GPKCDh4S4ejqP3r8y3Znh9To78APHCs5AUB-K8g,1219
flax/experimental/nnx/nnx/nn/attention.py,sha256=2qm-A1IkAbiqU5x6-C5fqODBPM8PLeNAzfrRc_2Lkhs,24533
flax/experimental/nnx/nnx/nn/dtypes.py,sha256=zEeCbEXsCeAFvGbwCR6gZx_KM70bQ_15S_uljb55aaI,3318
flax/experimental/nnx/nnx/nn/initializers.py,sha256=uOyOtYl4IbsPemyoRONbzHjj00pvvpSmmZecdaW_trA,2527
flax/experimental/nnx/nnx/nn/linear.py,sha256=09AOUA7B20YplPSTwXGkFKiaicO-787ABjeT_TG6ZR0,22638
flax/experimental/nnx/nnx/nn/normalization.py,sha256=sRl4e8fsx_enSrFszHnnFZPsTjF5vuh-fvkBOWvMgZM,13007
flax/experimental/nnx/nnx/nn/stochastic.py,sha256=XrvVzJyAo0T8i7nwwhLrbc-pHv216WNRA8ZK8NIBuAE,2905
flax/experimental/nnx/tests/__init__.py,sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg,1163
flax/experimental/nnx/tests/test_compatibility.py,sha256=7JGj2B3cc_TIOcNjaGTVQiIyfFYk2EaElZKr1cJIcOE,1204
flax/experimental/nnx/tests/test_containers.py,sha256=G6YVPA84zuPD55jXmJuxwVoWttF-jKqKESyxGqVooPA,2408
flax/experimental/nnx/tests/test_graph_utils.py,sha256=S8N5HtexOfHVViPHXLAo9yqQfK7I2LNSeq8TaaUeBb4,6431
flax/experimental/nnx/tests/test_helpers.py,sha256=a2FfI54avzM8TCLK4tmH79S_8QKaHl7edWkt5xYFEvI,2078
flax/experimental/nnx/tests/test_ids.py,sha256=iyzxi_ene5RGX7r9EBZl2XJTGQ3CEPZNsiLKqFeebfE,929
flax/experimental/nnx/tests/test_integration.py,sha256=GPckSz8KuTxzdqSKS4YZgoFckmzzEnjal6Kq4MgcO5A,7194
flax/experimental/nnx/tests/test_module.py,sha256=ZZRFOES8ELBq14jlO-JTs86vXFGqmkdk-dad1D6ZJqM,14401
flax/experimental/nnx/tests/test_partitioning.py,sha256=uhR6glWxt7-gnGfaDkADienyNQHtERUk-o9lOeOln00,4060
flax/experimental/nnx/tests/test_pytree.py,sha256=iTj1mpIwzR-PiA--MLXVdpLlZLTfBAKF8qD1S5o91_0,6261
flax/experimental/nnx/tests/test_rngs.py,sha256=u8zL85wKL_KMLvVbL0DUBiU8wCJiuFLdNcGB8kejpeE,4697
flax/experimental/nnx/tests/test_spmd.py,sha256=v0arqaGHhoeDIA9jsT5h4kyBqOiFOblPgh4_i2kYRiI,2130
flax/experimental/nnx/tests/test_state.py,sha256=wLF8TadLdpggPz_25FjlvDABkDtaLcWgdY3P1YecgJM,1977
flax/experimental/nnx/tests/test_transforms.py,sha256=z5O0hhLG01l8ha2iGsBiQ_FNydKcZgenHNNNNfhAXBs,17878
flax/experimental/nnx/tests/test_variable.py,sha256=HnNmpiD9Xwojl1ot7Egbfr26y0QUOvy5zK4Q-ao8Xlk,885
flax/experimental/nnx/tests/nn/test_attention.py,sha256=2g2V87rPlYv1jjDEG8FW_3nT41NbLxUE1pYOGMxcEMk,2875
flax/experimental/nnx/tests/nn/test_conv.py,sha256=3LBNvBxKuoEKCumccjjw8rbEmu7iYmbaTyIidTYxl-s,3038
flax/experimental/nnx/tests/nn/test_embed.py,sha256=qk2lJ7f4w2NJo7N-PYH91tKxypjO3VFt93nFMW8ME4A,2071
flax/experimental/nnx/tests/nn/test_linear.py,sha256=K4ObU_2WLOnAX3HqX7Ww1cGHFb0n5Fkx9k_hUn34KtY,2612
flax/linen/__init__.py,sha256=XUE8Hp4YC12wxtQSu7klt8IrJ1yWHOV8_ety7861W-g,5201
flax/linen/activation.py,sha256=LeFjjROwX7GLlyfdA-iBaRNY_qp1PKhGnir4ggxAVV0,4101
flax/linen/attention.py,sha256=Ac7zOJz71arRksoiZT6M5eOmlsXrFZgSWFIL2MSf41A,30877
flax/linen/batch_apply.py,sha256=7B_bwfI2vJyt2UPMtOcjEo6puLFavcJeLrF-COoAntc,4172
flax/linen/combinators.py,sha256=WW-GSBw19Wwv0Mpg_VOdYqSmH7-8loAqDkj5DJ1xcNA,3846
flax/linen/dtypes.py,sha256=hyXSAKgx6iK15cXrojy4f_DHzSq160Jdv_1Mg7vqoaM,3860
flax/linen/fp8_ops.py,sha256=FnMvjtBTILj3PpDkO857kBHq0i-qIHBJvel5dffw0O0,6687
flax/linen/initializers.py,sha256=CC4lY4uvAGisaxiiIAE2AHVPuN9q8EqQ2e9OZBDPi1o,2676
flax/linen/kw_only_dataclasses.py,sha256=U4-mGzgAhz902A1SL29J8Hdr33smxl8NUB9BwOz-4GU,8828
flax/linen/linear.py,sha256=7u3ec_Z6JuTfSxAZt25Y4tY3JGz48i2U42-4wg2JSL0,39037
flax/linen/module.py,sha256=2HDCvYd43rDoOYlIeL3uAnz5UsgNZQNm9sMrV3M-NDI,108274
flax/linen/normalization.py,sha256=glpEQv00-vhe4ec_atUCiGACBWhCFu1Ni9anp7qDSjY,48478
flax/linen/partitioning.py,sha256=gYtiQofDrg7ExL4awfZZsxq2VKiRixIVoY5pn_-nQAw,19167
flax/linen/pooling.py,sha256=x6BACBzHgMrCB3xAIRbp0bYmPL7ux_sIce83qvkggiQ,5554
flax/linen/recurrent.py,sha256=F8mlaDrfRslDxEYh79pzdB4v2ENpQakqF6PXKbNHPcc,41944
flax/linen/spmd.py,sha256=mZJKpZNMmlukrNd4GWbjMTHqF6CJrKOvyV0ecFL67s8,11160
flax/linen/stochastic.py,sha256=tUthbCWYuSfMW_dd397l_IRxDQyujsQ_4FRNcxSiGGY,3619
flax/linen/summary.py,sha256=FbQRmeGQZkBZ2KfKyBylKQLV4FGTI4sKzlZ53Wd02k8,26001
flax/linen/transforms.py,sha256=6msKCCI1nVMUZoFBUJooQLbzKlzK0yy1ONCf2hF7tds,78002
flax/linen/experimental/layers_with_named_axes.py,sha256=xv-vBWVgIB2lfHF4KrryM66jQkgjYgeC9qo2Ansg7K4,11235
flax/metrics/__init__.py,sha256=cmhQJ6zmLk9z5CZDNSdSASLx8PDpq1BW41BEvMVk0jg,1163
flax/metrics/tensorboard.py,sha256=8pZFFGn6jzW5-zemn5tCs1-NhgNQoZBgflmqoqEAW_E,7776
flax/testing/__init__.py,sha256=UarXpjywrTaGYJBYPf0joEzJawWavRs_f58YZJowfh0,647
flax/testing/benchmark.py,sha256=dumZs_E0VmGqiEQHwe8eOzxgv5bjWtXj5NKqrnsmF7E,9320
flax/training/__init__.py,sha256=uJTipyYX83XazXl5bkf17EF4DjJE5s1YhCA3vS0vXqo,613
flax/training/checkpoints.py,sha256=P56XV8DBqjm0be-UNLRlsIHotfw1iHIG2N9l4lNcLIc,43940
flax/training/common_utils.py,sha256=_3-gAQuFBC6ITUiu9dgGMZu0PS6SLlzd59UlnwfHAJg,3690
flax/training/dynamic_scale.py,sha256=YD1Aby3DqDlfi_KKkdVuW4gBi_rz0K2-3xIj1D5VHXs,6056
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=WfkaofHWR7S4jdOCQWrBkWb0tp0nV2KLGWmszunUuuY,3655
flax/training/prefetch_iterator.py,sha256=KnhVdWbGC--2YTaxKeuGJTfH2TrvtjQWP3mk3s6KHZQ,2978
flax/training/train_state.py,sha256=Hpm2kag2BtLh7PXoYP1xDCwo1vyb28bJXwiCt6vb3Vw,4731
flax-0.8.1.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.8.1.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.8.1.dist-info/METADATA,sha256=RDzV40Q59AP-AOks4-OmoSkEyiFdDCgY22XrDEMcU3g,10196
flax-0.8.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
flax-0.8.1.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.8.1.dist-info/RECORD,,
