flax/__init__.py,sha256=hBIbRVw50RwU0UMXN5cB-0lySabi9btEzkSm5jIPAoA,908
flax/configurations.py,sha256=cN99R2ZfujRd0I989n8x6yTmUQRLZZ_8zZtY9ud-Fdc,4168
flax/errors.py,sha256=JiM4dy2EbI2GN5o-qw8MDrfn4QD3_s_FCzchKecP2FI,29783
flax/ids.py,sha256=BE-pLfy_7Z0VM-QDbeGLr8dM_u9-txYXyyUw91bcyLw,1744
flax/io.py,sha256=qVFdnCfujDFR82zztjOBsKqvFEl53lM5sHQUm18VPC8,5055
flax/jax_utils.py,sha256=ZkXw3aE77QQPHFTo-f--UVB5_QsZvFjvbFRyh9o_MXQ,11461
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=Tl6aE_otMMXKov3ntEuR0OteFohWyq2XSECfVjB85Wc,14471
flax/struct.py,sha256=EKuNCrOGIOInbBs000SjWAZlcyFwbTr8iLzGTcGtF80,7330
flax/traceback_util.py,sha256=ozGNM-0mlncdvfQKXnqCXIcYpXzt0LLDs9tXUkGBnoY,1989
flax/traverse_util.py,sha256=BS7ZeHYQCYiVjCeUEggzyeRcsFhdNpbA1jPJmDPAf-8,13401
flax/version.py,sha256=bUfKhIjSAkTIY47U3aDWd7Zhf1yLg-_53kVkbjasPLY,651
flax/core/__init__.py,sha256=YIknvX8iqDIDmeCpUEr0th7UQvo0Rw608pfpxmIrFkQ,1352
flax/core/axes_scan.py,sha256=xjmGVOVqWOlLxeRkZ6p-VDXnWJDiZo5Q5wo8gmHRUaQ,5313
flax/core/frozen_dict.py,sha256=Q8Z4Z3XcOXbXFYHXJEkbLESlQlkZa97PnqiFHaaWwIE,7862
flax/core/lift.py,sha256=HFnxd8GTSZ2Mj3m1EpDag2qiZs33F4bw8JbAdYJXjvA,56619
flax/core/meta.py,sha256=v2O562c7Av3wmmbldVsP1JiS4tqE3O_sNDaOfa4amfE,10509
flax/core/partial_eval.py,sha256=lUxUlcP50u62piTk0sH43p38IdtzKfLRzN6v71MxVf4,2517
flax/core/scope.py,sha256=We7E_HvqVf31J-asyJgjrbcOJ1ASoh5cxxiSn3wqv-E,35706
flax/core/tracers.py,sha256=wekMcUklh-bBztaATP_9EWo3OJJdcek3AhuI8H-M2hE,1053
flax/core/variables.py,sha256=AVLWdjdHA9jVE0p6ny_gYpG1W3FcAvFulqqGhkTUII0,1558
flax/core/nn/__init__.py,sha256=otlsCGJ7E1nvx_G5bC2LnFGD3DE-bkUS7Fwoi3sFP-A,1744
flax/core/nn/attention.py,sha256=86MzvieYxaQl4UVhf2OJzaywV__XeG7KdYml5zbhFVY,18626
flax/core/nn/linear.py,sha256=p95-4GR30hyP9OI84uf29dPG1fUHWhn67yjAmdXt6kU,12538
flax/core/nn/normalization.py,sha256=h7UYFUdjCns4BuCUjuP1KGrXiONhxPhD3bKDBKkK2jQ,7164
flax/core/nn/stochastic.py,sha256=pswSFRyQws7CU_SWWY5NuNlK6lzt0ZcTcBRI1PDsKrA,1503
flax/linen/__init__.py,sha256=J22bVbAykPzbyYjUv8uH1Ntq4mwY0Qj5S3Er-qc3voc,3871
flax/linen/activation.py,sha256=wVvXgIg7Fc5-47ficEk6Ue35MxL1bPBrDL7FqPTrOKk,2392
flax/linen/attention.py,sha256=V-KXdF9X6kzmr6QA-Ou6qURQGPjDDl3pSIBj3h0rWIg,19132
flax/linen/combinators.py,sha256=Q5-tSe2DbcrB16HblPvvb4NH5ZRBb3TO_pxPOIA8Fu0,2891
flax/linen/dotgetter.py,sha256=4hBWVTtGx45rSuSC81GXeJXPyT-h6s2xtNg0EXoX2Rg,3383
flax/linen/dtypes.py,sha256=bGMUTwo9yq0l0syBlmxClwoMWJH4ELRwv0ws8zkJxz8,3920
flax/linen/initializers.py,sha256=JwrsqQI73aeh5ujFN1xM7Phox_q3JvjiwyobPQ7-zvk,2619
flax/linen/kw_only_dataclasses.py,sha256=87kNu1SbvtAg-83fl17r5hk5fedLDPRp8Njc4hKSI74,7528
flax/linen/linear.py,sha256=toqey_wblV3Fq7wcTOhuPMvaFuMfpJuqVmgRCc4XhRQ,32974
flax/linen/module.py,sha256=tQXAQkyHdc699-sNj1-QmrxhTQMAJxZpbOUu2FH6XNU,84112
flax/linen/normalization.py,sha256=U_ZnK9zZ0gELsmDNrQdru649t2oBlkmSzyHrNo2NW2w,21305
flax/linen/partitioning.py,sha256=FcZZTsVJ1G-FqBg-Y8qRr6E_VjJkMJUZwMHaKqZYtt8,19763
flax/linen/pooling.py,sha256=XldI1j4T4vUzcs-fY81xLQLxPqFzjqd25xmYVEw-lPM,5480
flax/linen/recurrent.py,sha256=UAupDzu-lMF6TJNcvupvBox-TP2mBCsmN7tx9dDcI_E,19138
flax/linen/spmd.py,sha256=UmQGVkHYse4lhd0NQbMdHKa6-w5BjmIa-Y8olJJBEdg,9967
flax/linen/stochastic.py,sha256=EsDJyTSSbrsacQCaaSw92KDSmQOFJMJZRZER8TkASis,2741
flax/linen/summary.py,sha256=seqQarVoGa_TWVPbhKHPqphOYwLE13qFJYASLBgvkUs,19177
flax/linen/transforms.py,sha256=IUFLUuY_OvxOXbAJpOm4GxXOTHI1Jf5lb6SrFfT-Y5I,58302
flax/metrics/__init__.py,sha256=YtkS-Bp6L4D50sa6m-MGVRb3ReK_APAL0UmEYY2yWLw,582
flax/metrics/tensorboard.py,sha256=cXIgKjrMWAVPntM5aAgFUw1-zQJXBl41ie-lV-ROXOc,7726
flax/testing/__init__.py,sha256=b3T0KBRRsHsHTIijaJKnMvGwSZs_8lk_OfZ_vTm4o5c,647
flax/testing/benchmark.py,sha256=Sq8ensY-TStG79qOETrykyyMELSRQfQf1HvKv28HGdw,9472
flax/training/__init__.py,sha256=dYve93CbiTJre88CKyORsXwF4xW2Do1e3i87Vq4gmlQ,613
flax/training/checkpoints.py,sha256=ogCDAd4ZHrg23aqVZBNHOhW1fXs7i4cU3GQzmRes32c,39574
flax/training/common_utils.py,sha256=tSfNPU5AaK46xWMk22LAIV1Fui_19p2fErmn9dz99mU,3678
flax/training/dynamic_scale.py,sha256=g5JeFmVWsWhaZB04HymUiNTMdpgZYvFxAUd_Ebnyt60,5484
flax/training/early_stopping.py,sha256=zlF0JiTf1WxOCh_OsfsdRyIjki8vqk2_2qy1hGzFgig,2718
flax/training/lr_schedule.py,sha256=JRIJA_b8OBD8nMD5eiXXnF2GClaSrOM_ZKOkpEPai58,7499
flax/training/orbax_utils.py,sha256=McHzbaNFeY3LiaN3KtzTV-RAqNEoWCgRJPD83kqDCPY,2078
flax/training/prefetch_iterator.py,sha256=7BvGUbpvuKATW1vpQyS99NZrMO8QUfTRNNlSxu54WJI,2971
flax/training/train_state.py,sha256=qG5TZxr_s-g0KPEkokf2JASNgwygkMotc_SvwRDy-V0,3256
flax-0.6.6.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.6.6.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.6.6.dist-info/METADATA,sha256=fDHwTZ9H4u6QGrrCi4ZMnLmRxasTg1lCVgHAUM6mCRg,9647
flax-0.6.6.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
flax-0.6.6.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.6.6.dist-info/RECORD,,
