flax/__init__.py,sha256=t2YdcVAZ3kbRn9ANPyE0K3vFwXsMKMpOHIt4sd1Ok3g,854
flax/config.py,sha256=wb4KhHwpDOFT0aYgmVAAzvfwy9Xk6PhsBXwPzcynnGM,2179
flax/errors.py,sha256=SSwy9zU1VQPznl8AswMUOHz4jMAVECv5Q3lDdleW060,26340
flax/ids.py,sha256=BE-pLfy_7Z0VM-QDbeGLr8dM_u9-txYXyyUw91bcyLw,1744
flax/jax_utils.py,sha256=2HPTb0zBx3S29QWr0gdS4rNNFTG_Ru9qOwLTQWeyfHI,12403
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=zsZhfCQPRbvJSMOvXpmKYiYauAyfxQLDUcoP9GMR1nA,13020
flax/struct.py,sha256=Mxncvydnq5d1S4-wmHHFo7vvxtzuB3qBJKscF0iFi10,7125
flax/traceback_util.py,sha256=ozGNM-0mlncdvfQKXnqCXIcYpXzt0LLDs9tXUkGBnoY,1989
flax/traverse_util.py,sha256=jGW_GvvB1zF5eypjmiIcfI7_9NaYVg7bjSkV0645ZkY,13479
flax/version.py,sha256=Q3YpjNxyEnygi7VeXeA5b4qMRH4b1LqyOG_T68KyUlE,651
flax/core/__init__.py,sha256=6UZESqwJQXDneeh7ext4NCcXF8_6DOdLR4aQQ-iGMuM,1191
flax/core/axes_scan.py,sha256=xjmGVOVqWOlLxeRkZ6p-VDXnWJDiZo5Q5wo8gmHRUaQ,5313
flax/core/frozen_dict.py,sha256=anmCWNRSBYfHPTUNLPTEHpIh6Zv4ou2B-4ayMOEJHfA,6305
flax/core/lift.py,sha256=N0mxm4LVBNisj2R6bpR9ReW2pclCdBB5trKSdfzg6SI,55239
flax/core/scope.py,sha256=KbGxiE3LAe13tyS9bvm225RxAp9ZFnaQjPLtSJxzQE8,31026
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=WAdfYF7DdT8jpP7R83R6dzyMZFduzJo_0ShgKf5oR5o,18589
flax/core/nn/linear.py,sha256=7F7RxW3v0kXJ3K9CPgYbG_FRQizzn_tEdl59hjoxPx0,12495
flax/core/nn/normalization.py,sha256=JwakdxeYaPeriDEQ5oX0RGYfQsQh4Nne44nAWwMLm70,7122
flax/core/nn/stochastic.py,sha256=pswSFRyQws7CU_SWWY5NuNlK6lzt0ZcTcBRI1PDsKrA,1503
flax/linen/__init__.py,sha256=MQxoH7iAW7h9FtnfHofcvNHvGOF3_Dpzinuk82OKWEM,3128
flax/linen/activation.py,sha256=GqZ3byvHEndEt-ud0sMZyggkR2DcKZno0GPABexZbak,2361
flax/linen/attention.py,sha256=1qIdmYkj7q5stSYr4fyEz453lDzClhcEhB3bFKmoLpQ,19133
flax/linen/combinators.py,sha256=rW49ao0SCMrHnaw9C76MTy5VTMkBx9qCCvBz3td1LkQ,1843
flax/linen/dotgetter.py,sha256=yyNlYDsjMNGK4ixMZOWx5v6ckJufms16uOOKt9StFXE,3347
flax/linen/dtypes.py,sha256=bGMUTwo9yq0l0syBlmxClwoMWJH4ELRwv0ws8zkJxz8,3920
flax/linen/initializers.py,sha256=4eyWi2qwl7fKWXcqv2B981kZjWzy5XdhtkCIgk7JpOs,1785
flax/linen/kw_only_dataclasses.py,sha256=87kNu1SbvtAg-83fl17r5hk5fedLDPRp8Njc4hKSI74,7528
flax/linen/linear.py,sha256=2-o1r0pMXa2CkB2Zx6CX2kE5LjAr17TzS4PDJoeW2h0,31818
flax/linen/module.py,sha256=zsjfnPk06ZMJALLnSNIi_GyOJQZPWXepSiwVisYQYTg,72443
flax/linen/normalization.py,sha256=PqVLNk_xW-f4oL-__vC6CAjWan2AzgaRyVyLMJrWPqY,18097
flax/linen/partitioning.py,sha256=Z5LHpjS94SRuOu2aBoLraPToqrNgM_CiK4_T1nziZ6s,25280
flax/linen/pooling.py,sha256=0iY0GGEToCraONKOAMNOeYUH1eid8lUfYqa1B98u6cA,5346
flax/linen/recurrent.py,sha256=jHFzZAsPbELxnHWm_qBGdRkSt2K8n84KalHDr8oxVX8,18754
flax/linen/stochastic.py,sha256=EsDJyTSSbrsacQCaaSw92KDSmQOFJMJZRZER8TkASis,2741
flax/linen/summary.py,sha256=4_BP64vvxicCV-IPQbO7NEmbm0AzGT69616oaTgzBXk,17207
flax/linen/transforms.py,sha256=Y0rD236nxtny37br5WbZb8bOO3J4idbEXXDfPyUKiOE,54323
flax/metrics/__init__.py,sha256=YtkS-Bp6L4D50sa6m-MGVRb3ReK_APAL0UmEYY2yWLw,582
flax/metrics/tensorboard.py,sha256=E_TlTTvKdtTJf_d2m1gemjyVJpDKVmDSNEUv7FzH3qY,7734
flax/testing/__init__.py,sha256=b3T0KBRRsHsHTIijaJKnMvGwSZs_8lk_OfZ_vTm4o5c,647
flax/testing/benchmark.py,sha256=tNS32jkvnlIb7qVAaqJXXJc3Y_0mNiRmNyRInqUXrOo,9521
flax/training/__init__.py,sha256=dYve93CbiTJre88CKyORsXwF4xW2Do1e3i87Vq4gmlQ,613
flax/training/checkpoints.py,sha256=eWF-aWCJXFY2FR27GiXknwpJvkvD1FYm0oLAncgr-wg,33559
flax/training/common_utils.py,sha256=tSfNPU5AaK46xWMk22LAIV1Fui_19p2fErmn9dz99mU,3678
flax/training/dynamic_scale.py,sha256=yoczqaME72t5UhoIs0hJx-1MH95Ccj2C--lJDWPQSro,5482
flax/training/early_stopping.py,sha256=zlF0JiTf1WxOCh_OsfsdRyIjki8vqk2_2qy1hGzFgig,2718
flax/training/lr_schedule.py,sha256=BICcOGRviI_d3wwRB4foIcEB9Ggfo3N56mX6Tt3NueI,5413
flax/training/prefetch_iterator.py,sha256=7BvGUbpvuKATW1vpQyS99NZrMO8QUfTRNNlSxu54WJI,2971
flax/training/train_state.py,sha256=qG5TZxr_s-g0KPEkokf2JASNgwygkMotc_SvwRDy-V0,3256
flax-0.6.2.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.6.2.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.6.2.dist-info/METADATA,sha256=lrMGwDLdFt2LKrEz5plysHyBuB630GE3h7_8zIqGM7I,9477
flax-0.6.2.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
flax-0.6.2.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.6.2.dist-info/RECORD,,
