flax-0.4.1.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.4.1.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
flax-0.4.1.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.4.1.dist-info/METADATA,sha256=HqiVQtML9ZvnYivYVocGPOZDLV0X5gDo0H3kbPNRGDA,9126
flax-0.4.1.dist-info/RECORD,,
flax-0.4.1.dist-info/REQUESTED,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
flax-0.4.1.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
flax-0.4.1.dist-info/direct_url.json,sha256=FTogPhlwjlo6c3EMFJ46OsrAQh1tHXqjlV9xajSluNk,100
flax-0.4.1.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax/__init__.py,sha256=IS1U8vdWd1mwkOdNoewfA-BPkl4AIENfNlbZkWisnPw,861
flax/__pycache__/__init__.cpython-310.pyc,,
flax/__pycache__/config.cpython-310.pyc,,
flax/__pycache__/errors.cpython-310.pyc,,
flax/__pycache__/jax_utils.cpython-310.pyc,,
flax/__pycache__/serialization.cpython-310.pyc,,
flax/__pycache__/struct.cpython-310.pyc,,
flax/__pycache__/traceback_util.cpython-310.pyc,,
flax/__pycache__/traverse_util.cpython-310.pyc,,
flax/__pycache__/version.cpython-310.pyc,,
flax/config.py,sha256=ENL8un8tvDpAKQQc0rQpcAZfvgQWPWaWJ09xDBv9E4Y,2101
flax/core/__init__.py,sha256=t8z_GdNAU-Ttt-YN6PKTm2bJ-PU3SqnFmZ4pGAEvLoE,884
flax/core/__pycache__/__init__.cpython-310.pyc,,
flax/core/__pycache__/axes_scan.cpython-310.pyc,,
flax/core/__pycache__/frozen_dict.cpython-310.pyc,,
flax/core/__pycache__/lift.cpython-310.pyc,,
flax/core/__pycache__/scope.cpython-310.pyc,,
flax/core/__pycache__/tracers.cpython-310.pyc,,
flax/core/__pycache__/variables.cpython-310.pyc,,
flax/core/axes_scan.py,sha256=0NoNa1WIGGxEWn8JXihxms7bqa7ZcRcTAjeu6kZzRJo,5109
flax/core/frozen_dict.py,sha256=JPjqEjEd6Uc-fhtgBTaks5p9IaAkQ4DpDtPHhCuin4I,6285
flax/core/lift.py,sha256=M04o3FLIIw5q7MmCqjpDqw04F_6TfxtcLEgGQlaL3oI,49239
flax/core/nn/__init__.py,sha256=nw4Td51jARhfNHUw0B4aQLVSG2qjdqZmxX1CGgP836s,1378
flax/core/nn/__pycache__/__init__.cpython-310.pyc,,
flax/core/nn/__pycache__/attention.cpython-310.pyc,,
flax/core/nn/__pycache__/linear.cpython-310.pyc,,
flax/core/nn/__pycache__/normalization.cpython-310.pyc,,
flax/core/nn/__pycache__/stochastic.cpython-310.pyc,,
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/core/scope.py,sha256=d8rJmIKw51E_Ri0f_aEb4FVuugGte5GYV7hCf7I2t7E,28902
flax/core/tracers.py,sha256=xzAymR9fu0fPENJcQ7y527zsQmPaed2SWyH2T9nKT8A,1054
flax/core/variables.py,sha256=AVLWdjdHA9jVE0p6ny_gYpG1W3FcAvFulqqGhkTUII0,1558
flax/errors.py,sha256=65gYjE1mOY-iQfp0XO3Fz80U3YXmWDTlcBksfWg7ZEQ,21114
flax/jax_utils.py,sha256=5qkcnsnFsN0kCBGEsCwDx04YdX5IPvIGZb8dHX7mrjY,8884
flax/linen/__init__.py,sha256=X4-oKzT0oXGxWUTVLmjpZzv3wUJxVoWWnLbr6cnpre4,1986
flax/linen/__pycache__/__init__.cpython-310.pyc,,
flax/linen/__pycache__/activation.cpython-310.pyc,,
flax/linen/__pycache__/attention.cpython-310.pyc,,
flax/linen/__pycache__/combinators.cpython-310.pyc,,
flax/linen/__pycache__/dotgetter.cpython-310.pyc,,
flax/linen/__pycache__/initializers.cpython-310.pyc,,
flax/linen/__pycache__/linear.cpython-310.pyc,,
flax/linen/__pycache__/module.cpython-310.pyc,,
flax/linen/__pycache__/normalization.cpython-310.pyc,,
flax/linen/__pycache__/partitioning.cpython-310.pyc,,
flax/linen/__pycache__/pooling.cpython-310.pyc,,
flax/linen/__pycache__/recurrent.cpython-310.pyc,,
flax/linen/__pycache__/stochastic.cpython-310.pyc,,
flax/linen/__pycache__/transforms.cpython-310.pyc,,
flax/linen/activation.py,sha256=zUvAJ0ICA9llkp-2dozln1c23qivNK2JKeMgy4BoOAA,2113
flax/linen/attention.py,sha256=UjtQqvnGYvKF1BGvObtY-bGKuI3ox8XBnkQBZGsbTYk,18753
flax/linen/combinators.py,sha256=rW49ao0SCMrHnaw9C76MTy5VTMkBx9qCCvBz3td1LkQ,1843
flax/linen/dotgetter.py,sha256=yyNlYDsjMNGK4ixMZOWx5v6ckJufms16uOOKt9StFXE,3347
flax/linen/initializers.py,sha256=WLa5jktt-lfjVydjqg_IOowxwS8aTE7EsjRq-TbhMqw,1514
flax/linen/linear.py,sha256=-jIFxYbdH-f_or4sCLvzNGjMKt76Team3Oo334GKhAQ,23206
flax/linen/module.py,sha256=rlhTHpB7Bp0MztjvS7dGo54_47sdWLOcqkDhu4avUCg,58324
flax/linen/normalization.py,sha256=QGq1tKDfHmMsTFJvHu781DSkLHe4pE5x2l_rPpW7xUc,16145
flax/linen/partitioning.py,sha256=oZQeg-kmwt-3EuFa3TV-vF-05PZRyzW8qLthEPPEuZ0,22693
flax/linen/pooling.py,sha256=OzlX3RNvdQ0D37VMGK5PQA6-g2hRSjNWFsD0ncwfDQo,4940
flax/linen/recurrent.py,sha256=d3Y1HXMDVw0duH1n6LN6siDbC-Pf5GVi-kARhZMN7RI,18642
flax/linen/stochastic.py,sha256=WUTIN9BS29z-DyopB83CA9916rWuxjfSTvrW_qZQDpY,2313
flax/linen/transforms.py,sha256=Rop-CiRrQG5ElV0b-WJDe8unNmZ-ky-oseRz-FRny5M,48531
flax/metrics/__init__.py,sha256=YtkS-Bp6L4D50sa6m-MGVRb3ReK_APAL0UmEYY2yWLw,582
flax/metrics/__pycache__/__init__.cpython-310.pyc,,
flax/metrics/__pycache__/tensorboard.cpython-310.pyc,,
flax/metrics/tensorboard.py,sha256=2aJKUhYXFeTxoFs1ksMVv_-9m0PQzIt95wz9By1utPY,7011
flax/optim/__init__.py,sha256=iwP4mIU3Jtetq9I1LvICNzXLWBfoPZVQN2bBZuPXA3Q,2025
flax/optim/__pycache__/__init__.cpython-310.pyc,,
flax/optim/__pycache__/adabelief.cpython-310.pyc,,
flax/optim/__pycache__/adadelta.cpython-310.pyc,,
flax/optim/__pycache__/adafactor.cpython-310.pyc,,
flax/optim/__pycache__/adagrad.cpython-310.pyc,,
flax/optim/__pycache__/adam.cpython-310.pyc,,
flax/optim/__pycache__/base.cpython-310.pyc,,
flax/optim/__pycache__/dynamic_scale.cpython-310.pyc,,
flax/optim/__pycache__/lamb.cpython-310.pyc,,
flax/optim/__pycache__/lars.cpython-310.pyc,,
flax/optim/__pycache__/momentum.cpython-310.pyc,,
flax/optim/__pycache__/rmsprop.cpython-310.pyc,,
flax/optim/__pycache__/sgd.cpython-310.pyc,,
flax/optim/__pycache__/weight_norm.cpython-310.pyc,,
flax/optim/adabelief.py,sha256=CisOe9k4iAJ76OiQ7MbFAIidCFLCbvegrqVa1_Zcb_o,4420
flax/optim/adadelta.py,sha256=VYtolyzMI0IDY0eLORBpTOdX032CtwVzMeJkiiTLsTA,2586
flax/optim/adafactor.py,sha256=iefr8bmfOkCnETVL9rRx7MIG1tC2chehEiEuG6EaSF8,7872
flax/optim/adagrad.py,sha256=IE2PCRPx-ZpPf3cCbyT8HWuS1l7KhpZKhVJXXIo0U5k,1837
flax/optim/adam.py,sha256=qv3Gek9W33FDgLBv0Zb_PEDugosUW3qUPH4gYGQxbck,3432
flax/optim/base.py,sha256=66VV-u6EM-cesJx6uuszjXUdBqkdDaQPeS__NInSx08,20558
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=eN_VXwV8i_b1y5d42HSRNJ4ADv0-_TWuLXDap1mMyp0,2764
flax/optim/sgd.py,sha256=-xO-5ij9HQ-_SvgQz3xJigKfeJaQjA_oRNCmrJB5OnA,1399
flax/optim/weight_norm.py,sha256=7f8VR94ILO-frewlPVQ2p4jpWBTP0W4Oi_M7rHYt01Y,5376
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=O_5OaVxbq0tgyL6MORNADSrFgc9birRY1AUO9Dvrs9k,12908
flax/struct.py,sha256=LBxh1X9SaGd0xcV3hJNOrvdUxhL2PC92KmFTa4bI9N4,7606
flax/testing/__init__.py,sha256=hKWEbKmPH84atZdxK30BEo7K4kK7IqLrCizoC_s3Uiw,667
flax/testing/__pycache__/__init__.cpython-310.pyc,,
flax/testing/__pycache__/benchmark.cpython-310.pyc,,
flax/testing/benchmark.py,sha256=1CSZQ6vf-KLCmhrHYcxpKrakfcaWQw9hPVbGdiVnA_A,9562
flax/traceback_util.py,sha256=ozGNM-0mlncdvfQKXnqCXIcYpXzt0LLDs9tXUkGBnoY,1989
flax/training/__init__.py,sha256=TE21K5jkA5ZP28dHY-CtjfSeVMb4i3a4XUw2GynSnNc,633
flax/training/__pycache__/__init__.cpython-310.pyc,,
flax/training/__pycache__/checkpoints.cpython-310.pyc,,
flax/training/__pycache__/common_utils.cpython-310.pyc,,
flax/training/__pycache__/early_stopping.cpython-310.pyc,,
flax/training/__pycache__/lr_schedule.cpython-310.pyc,,
flax/training/__pycache__/prefetch_iterator.cpython-310.pyc,,
flax/training/__pycache__/train_state.cpython-310.pyc,,
flax/training/checkpoints.py,sha256=hAavTI3R8WI2qlMQAvKjaAW90pbmjJ8kYmo5oFiKC_M,12685
flax/training/common_utils.py,sha256=egnGTb959oQ0fK0lloHY18tY4WAbZbFNLtUwN9_Zy08,1899
flax/training/early_stopping.py,sha256=CRGHQorzreGUcYGmxegJc27zhftqaPCs6WN98BVEwMY,2034
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/traverse_util.py,sha256=2j7XXVhC7tD4xCfaABEj2Sh7FTNKYZjrdgVQBoGjD5o,11734
flax/version.py,sha256=beZ-6JXLY-7kuerx0H2UiKw_mV6WdmyK3w5MbrAzc00,651
