flax/__init__.py,sha256=bKCVMzasK_shva-oFrzrg5wIFifPPLZe_FcYTAgWt0g,908
flax/configurations.py,sha256=218Iy5FQzos0OhJ1bwrRUu1qWxpvHRKz7u9uIGdoAww,4168
flax/errors.py,sha256=RKUkb6PMcIumPwa2nzX-tVGr6mBt-1I08a_a2FDo8Y8,29793
flax/ids.py,sha256=DCP11EoJKs8DYcT-B2RzD4auyuTI1UW-ZN3hi4xQb3M,1744
flax/io.py,sha256=nBoYPKpchpEUiazJaCCwVNnmQM0G6I-QQYzTIc0r1X0,5316
flax/jax_utils.py,sha256=IXX2b9ikYn922uktNphXSirCEA-9mOYiozqXYqzWO9Y,11483
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=8JssktBFZKZk-iW_dj2Csv3_MMpRY82uAIjvpbQmfu8,14471
flax/struct.py,sha256=e8RWEJJ6xeh82Jt6KzE4SThzOYkF5bwaquyEsoACTL4,7332
flax/traceback_util.py,sha256=71vyqPSTJfPUSbmz_INTuvYk9I8vSnachAVinXpMq1k,1989
flax/traverse_util.py,sha256=87U4-ubf0OE26tJys3O2itYWDBwsj1qszWUeKrHl8Fo,13401
flax/version.py,sha256=NQv2M4sZJ-6IGBhtQ0ERaR1tu1KPe4OEMd52Eht0hY4,651
flax/core/__init__.py,sha256=Rh4WMIPr7PCiqZUeyL2LRVH-qvWkfy8N-aigAwsoTbU,1382
flax/core/axes_scan.py,sha256=Q8qUpWvK-hfcF-68JJxnJTSMKktRFKs6hsaszwZH0Kk,5336
flax/core/flax_functional_engine.ipynb,sha256=Fm9guf0xme8voulyRpqhCyWWyir4IUNcZOetCDTBZjo,10412
flax/core/frozen_dict.py,sha256=QDzJs4Z29f3G6onZFHVcVqrKTVXLFSyzut9N4I-3EvM,9418
flax/core/lift.py,sha256=1MZJc9L5zr1y-sq3Zp8Y8UP1Wk0jQuHs8uUuQqqcGIw,56619
flax/core/meta.py,sha256=_qeZH8pxewuzPqEU4X8MgC7Yaa_uOZLZ769gaOtjySc,11617
flax/core/partial_eval.py,sha256=k5h4ZFXsIQBcvesCSuL_9XR7PWReNRq8nhQI3kY8m80,2539
flax/core/scope.py,sha256=_-v5b_SnR6bsqRZoh1Fx6pb6WTkvUUt17rPMdevbO0s,35706
flax/core/tracers.py,sha256=Oi656iLKyEK0U1xXNhLFCFTnO2x6hn-FUmOqoot-sBw,1053
flax/core/variables.py,sha256=ki6bPJsAtuUNSuQGMgGcv67a4pq8cALK4C08QDQ48Ao,1558
flax/core/nn/__init__.py,sha256=cqzMErC40qveiTeGNnqQr4BvYz2mmxww0JlgTqPfuHs,1744
flax/core/nn/attention.py,sha256=rCEOPs8qom9wLU83kmSz6V8SASaEqnUv0TaloCBWYlw,18626
flax/core/nn/linear.py,sha256=ch4MN5JL6DsboHKPod6qnItDiPa8KS_UcfEzh7JSuxY,12538
flax/core/nn/normalization.py,sha256=OVkAMA4mven_tUkDa1UAdeHenCgxPihpe_FjfaQaftk,7164
flax/core/nn/stochastic.py,sha256=IeAleelNp1JwUYc6P94NA5cNQ10ckZeZW5dEKI9TLWA,1503
flax/linen/README.md,sha256=LrmgiMP1AI5cS3SjYtJEIkiULNMec0Xe2JZzQFGwgkE,2195
flax/linen/__init__.py,sha256=eAs8YydEvnwl_wev_XUvTGPZnpMTw9lhqoN5dQl3mdM,4014
flax/linen/activation.py,sha256=fNNPINqYVmW6dEU9Jkf1Q_5cgOMxKW24vB8sDENu3Y4,2392
flax/linen/attention.py,sha256=ShG6rWRDObGTbcXncIYAMZgPHqCS4-GUlTiESq6sIFk,19157
flax/linen/combinators.py,sha256=2EXw4FSa16X-1bn2rKbXsU0rOrguY5nlxGXFfrV9koQ,2891
flax/linen/dotgetter.py,sha256=I-NiDpOHMNIaPpzTKcq76XpJwgYeY3Ix6JjzBOoHIjE,3383
flax/linen/dtypes.py,sha256=TRxsRmVLaxJ4n1-ozg6iuQ93TpsP8ZWq_JVQxT-vKd0,3920
flax/linen/initializers.py,sha256=sXDiRSL0ACHNheGW04ZR3zxQsXlVtklX5qWxnf0aXx4,2619
flax/linen/kw_only_dataclasses.py,sha256=WCvexQdu4Yf_pBg3BfzRfTjB8lr2GLmFxelIA6h4YSg,7528
flax/linen/linear.py,sha256=NiEOcRYAGag_QM5kHDphN71eUBwkJb8bmfgUJ6uJztE,32979
flax/linen/module.py,sha256=98QD_RB8hy7JzOw6sBIgQHnlThZeULfVGizNf9g5Ts4,86133
flax/linen/normalization.py,sha256=CNt1ckcJm6BBoJMmR0cO4xIB1bXGi9M3TXF035soya0,21305
flax/linen/partitioning.py,sha256=xdbgNA81aWSpvIDtuO2KJUx5AmKCMwUztxuDkdR7sbc,19763
flax/linen/pooling.py,sha256=HC-XgUXCthVBqaM_HwFAMGih_xuYbPW-SAEGEpN_818,5480
flax/linen/recurrent.py,sha256=hXfDL-VMxuNJbvjimeDYSMAqpYpWI1zpoxF9v2LbQ3M,36844
flax/linen/spmd.py,sha256=GAi_tEj9fI-yA6TFMt1XcrqxGC1PLzV-dIHY9sm9pyM,11368
flax/linen/stochastic.py,sha256=5q6V_Z_KBk7YudlJ9dynFC7yDyzM_cCdLKixk3XV3GM,2741
flax/linen/summary.py,sha256=vFTaM7qOmRF3wqJNwDncCe_ALYQAP-ZYLJ3rzKrKuCM,19162
flax/linen/transforms.py,sha256=59g-e4EilG9c236qgY1SmB3KTcFz3_tj3ZxOV-UDX5Y,59698
flax/linen/experimental/layers_with_named_axes.py,sha256=nnwQvFPrRZ6-3_oizhGz6ByuNC7rlj4T8B8fFiQ4wO0,11542
flax/metrics/__init__.py,sha256=Z5yTcf44Jfp4-a6XUiIMTRlBUmA6WlzgbffCc4SRycg,582
flax/metrics/tensorboard.py,sha256=nXcCIV7EWGpS8W2F2j3KTawHLRvAR-dKVwmbXHKVfU4,7726
flax/oss/ .git-blame-ignore-revs,sha256=eJmLQj4P4rVEN8uRkflxwOhLFte29Y0WeGoI1uRNWfY,443
flax/testing/__init__.py,sha256=_OTLJJ6QRBv5CH6a3oK8MpUjEwd_OnMyzyvC6HleIgM,647
flax/testing/benchmark.py,sha256=abycb3FQ4atjIU9NUyYWHtnHPbSCqngVvXQNVbBDvus,9472
flax/training/__init__.py,sha256=wXT2B9npbmhWzD-dGXNkWQwDU-JpoUxOni-qiZVP0KI,613
flax/training/checkpoints.py,sha256=_mhcPA9V_zNivirEphpDbN29zbfbgIrrTpacRNl82sY,43229
flax/training/common_utils.py,sha256=Pa2sywhA4PA5bheLQYurByHEHPEQh2QyeaidjsQ-YS8,3678
flax/training/dynamic_scale.py,sha256=OKRnDxOI0Nd5ZWQgQNYTk3SPJEmz1GUiULtM_9c-GPc,5484
flax/training/early_stopping.py,sha256=HwoPTQ4EVh8CN9d78-P4GF5KBwx6FipCz47YKWHVSAU,2718
flax/training/lr_schedule.py,sha256=Rn48w-YZ3czJ22E1OC-ej7xKUU1iWubLaeX1aCNSiU8,7499
flax/training/orbax_utils.py,sha256=gZbsbmRB7FSy6Rnnp-3YPufHMUdycn4fwv23GEgowko,2694
flax/training/prefetch_iterator.py,sha256=p3-oghX_nhSDm_zejZb_ApkfPLI90TNxFeJ163eIky8,2971
flax/training/train_state.py,sha256=tERwq5Nuv6zmAltZmpOEF6Ja7gv--FsvEBvUGvSGLsg,3322
flax-0.6.9.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.6.9.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.6.9.dist-info/METADATA,sha256=HY1AZRvrupbrbC5fJvuK7j67S1zEPvmEMlaFe6oBe2E,9907
flax-0.6.9.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
flax-0.6.9.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.6.9.dist-info/RECORD,,
