flax/__init__.py,sha256=Ep8zivubhpQa9kTxPMuHC_SsZAzeAP-X8Qi0EPqmrGs,867
flax/config.py,sha256=L-Nty3JpgieohtqbV87uBDxW6pqN72eC7NIyCO3BAps,2247
flax/errors.py,sha256=ySt4uCq0lVJj8GWYEm1ZLkYH3NON0O7THpvpmKgPdgA,21275
flax/jax_utils.py,sha256=UaSxa5qZBArEnUCrJyVrXBQNRymdDvF4zUPpDbrZ3vI,12187
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=O_5OaVxbq0tgyL6MORNADSrFgc9birRY1AUO9Dvrs9k,12908
flax/struct.py,sha256=Ks3JVsW_R7A4-r9Vq_F7F5eXjMAeJdFpKE-R4vyGKXs,6986
flax/traceback_util.py,sha256=ozGNM-0mlncdvfQKXnqCXIcYpXzt0LLDs9tXUkGBnoY,1989
flax/traverse_util.py,sha256=8QDWNLUfSjiqhuBDVcAelqF4YvI6gwHZ2CBK_TVLSjk,12307
flax/version.py,sha256=M46CwcQ_nozm3cc4LVRY7N9SVZkYOlE9P76IQ5e5u1Y,651
flax/core/__init__.py,sha256=6UZESqwJQXDneeh7ext4NCcXF8_6DOdLR4aQQ-iGMuM,1191
flax/core/axes_scan.py,sha256=uw9zkBZNjWPLUc5vBCytRGoKG-hviX4nwmkjptf4hlk,5087
flax/core/frozen_dict.py,sha256=JPjqEjEd6Uc-fhtgBTaks5p9IaAkQ4DpDtPHhCuin4I,6285
flax/core/lift.py,sha256=gZxfV9aleuqsumzwYC7cGrcWQ8A1UE8kjqk18ERvH4c,51584
flax/core/scope.py,sha256=7uV-nAaafuWS2Nefl30FTnyZSzDiMrV4sGmlHR1H9Lg,29715
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=wXrIMwI9geH9pCFRvZGCJ5-jAGgQPDuZcNVLWinlkxk,12501
flax/core/nn/normalization.py,sha256=H7hQ17Yex2P3Ync-P2BlGB337hgziPzcxptdWUpk43M,7124
flax/core/nn/stochastic.py,sha256=pswSFRyQws7CU_SWWY5NuNlK6lzt0ZcTcBRI1PDsKrA,1503
flax/linen/__init__.py,sha256=Kyl7Z0mHPw0jiyEA7z6HAl_ESrLYz4_ef9WaS_6Tp1s,3077
flax/linen/activation.py,sha256=GqZ3byvHEndEt-ud0sMZyggkR2DcKZno0GPABexZbak,2361
flax/linen/attention.py,sha256=8Kn3lHIbXZzSqMZwe5ACkN4fzp6MN-Bfjii2N9QuALw,19128
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=WLa5jktt-lfjVydjqg_IOowxwS8aTE7EsjRq-TbhMqw,1514
flax/linen/linear.py,sha256=rABOP7lbgpuj5G-5PqbQ6slGh0XcE6LVtNBJk7n85T8,24624
flax/linen/module.py,sha256=kndi9Xf9t48__6tQTpHbOYCWSEkUYGePG70Rr-WCg5M,58258
flax/linen/normalization.py,sha256=WMNWF1XklOO83miJbNfZF9LW7PBjR8yNyzVG_CXed9U,16699
flax/linen/partitioning.py,sha256=4jTYY8oarw_LyelnH0MHOEH0J5qOVAnIO6A_1dYmUG0,22963
flax/linen/pooling.py,sha256=gQH15feyvHw874ExvRRkdJsICv-37UW3-zVRctxRnHg,4957
flax/linen/recurrent.py,sha256=s3WKf6r_u9w6009l3HQUgZJndaAn0CzXYQbnkqd_USk,18746
flax/linen/stochastic.py,sha256=ezDw332YEfFHiPfQ59e1oUb45m-VGFkx-EIs_YiPjBw,2420
flax/linen/transforms.py,sha256=xDuELWEVeGpVsMmzhpQAkxJgy9w-wPj7fKttq-WARyc,51479
flax/metrics/__init__.py,sha256=YtkS-Bp6L4D50sa6m-MGVRb3ReK_APAL0UmEYY2yWLw,582
flax/metrics/tensorboard.py,sha256=71a-TkTY-CvK55V3Ie88yisavuGSVg5b5nhH4iv2tOk,7043
flax/optim/__init__.py,sha256=iwP4mIU3Jtetq9I1LvICNzXLWBfoPZVQN2bBZuPXA3Q,2025
flax/optim/adabelief.py,sha256=CisOe9k4iAJ76OiQ7MbFAIidCFLCbvegrqVa1_Zcb_o,4420
flax/optim/adadelta.py,sha256=URmZwINtYBtECCfnejG7ye5_kUk7vAyvgP57tMdSrsk,2624
flax/optim/adafactor.py,sha256=iefr8bmfOkCnETVL9rRx7MIG1tC2chehEiEuG6EaSF8,7872
flax/optim/adagrad.py,sha256=9wvP5AV-3srzS6JgZVig_atqM72tI0yqxTNOZw_fiCM,1867
flax/optim/adam.py,sha256=f6OMC2Ks9LSVII70XORJz09APh1LhKxjaWJEEEDMFIw,3870
flax/optim/base.py,sha256=T6c9KDhayaumIPjTQv_QXixs4Wf1EBD-NnQgAyxbCXE,20572
flax/optim/dynamic_scale.py,sha256=4vuz8BEsdI6kVZ9u6CM7HYaolaRgyjC2yqiSJO8V26k,5409
flax/optim/lamb.py,sha256=yPNPCGi5-Hcbfty_31dw-E7C-qgcbCzeaVCQXbCrN-M,2987
flax/optim/lars.py,sha256=Rzm4f8R0U-r3tkD4uuVPQkQP5zr5uY8jRYwN6Q3f-Lg,2957
flax/optim/momentum.py,sha256=PV5EyFP0rC_jhwsJVQi1bJA4ht2VfCYZrWmJoyhQoBE,2231
flax/optim/rmsprop.py,sha256=YjoEdKGKd_BwtXBmOvi8rjj6n7cBcyar9syDBm-P22w,2801
flax/optim/sgd.py,sha256=-xO-5ij9HQ-_SvgQz3xJigKfeJaQjA_oRNCmrJB5OnA,1399
flax/optim/weight_norm.py,sha256=nNWhj13WiOzeFqse8WiypFrRxbARtLycYR_9rwUzWxc,5356
flax/testing/__init__.py,sha256=b3T0KBRRsHsHTIijaJKnMvGwSZs_8lk_OfZ_vTm4o5c,647
flax/testing/benchmark.py,sha256=1CSZQ6vf-KLCmhrHYcxpKrakfcaWQw9hPVbGdiVnA_A,9562
flax/training/__init__.py,sha256=dYve93CbiTJre88CKyORsXwF4xW2Do1e3i87Vq4gmlQ,613
flax/training/checkpoints.py,sha256=nJ9P6-w5rOMOnEg-PiCvZQOrQjpqc2ZcqLnIi19U24c,12771
flax/training/common_utils.py,sha256=ynpX74j7RrSMzJRhhVWiEx6NduZx6X1PESh2bHLsm3o,1894
flax/training/early_stopping.py,sha256=CajOw743juCQpog7Y37lrITTuV5VtzkE5jMHJfVDJrY,2725
flax/training/lr_schedule.py,sha256=iSHOnRf9xEarEsjeQfVUv3XLg0cvHY6bQH_Z3N0DkoM,5414
flax/training/prefetch_iterator.py,sha256=7BvGUbpvuKATW1vpQyS99NZrMO8QUfTRNNlSxu54WJI,2971
flax/training/train_state.py,sha256=qG5TZxr_s-g0KPEkokf2JASNgwygkMotc_SvwRDy-V0,3256
flax-0.5.0.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.5.0.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.5.0.dist-info/METADATA,sha256=SfbYXB4Z5xlFVqvJMm7EYqg6iK7IExFB3J-i3uMiYiY,9334
flax-0.5.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
flax-0.5.0.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.5.0.dist-info/RECORD,,
