flax/__init__.py,sha256=z3xf8PNJWlhPjbIaVMOHwT4gHDqHoTq8-5bNsG7duwA,1316
flax/jax_utils.py,sha256=rn0j9w5MzJ31Xsaj_Dj1By4iZF8eQ8dR2Xi8s9Ctp_w,8480
flax/serialization.py,sha256=Vnp4NZATgbmkvkIdDvU4Y04wMM6c7xj-oUTswWHIBK8,9726
flax/struct.py,sha256=YfqLK3j6HpN67Q76k5i4qQcG_gLNSTBifCAjz-WfTIs,4675
flax/traverse_util.py,sha256=dTjhZ70HXf9w1mq7HkbHLOKoo4nUxes6NoVksaZzknM,9027
flax/metrics/__init__.py,sha256=wFEwierDxa4l9XWaIfHCXV28dWDubsNvHtGCxGOzjPk,582
flax/metrics/tensorboard.py,sha256=OajSladfuxG217y9LBLPQHcdXgLDd-syKysXtOWiRTM,5539
flax/nn/__init__.py,sha256=lvssKKnl7BNgGOnU-rTk_I8aFuKgjRjUaqdtZGmLT_Q,1452
flax/nn/activation.py,sha256=S57dBhipGG49OrCzyBSQshpXic7TBonjMyOOieC810s,1142
flax/nn/attention.py,sha256=bVLnduqT4f_1m1nroXKMzyY8a8KM5jWo1I1Wmt2miZ0,19776
flax/nn/base.py,sha256=qFNLBc7LiMPEFwVDbh5UMv4ypyE0hcpTy9k-Y7oEk50,37509
flax/nn/initializers.py,sha256=spNjX9GSO6mIJr7AIg3gH0k9jGrMXWl1TUBudn-ohQk,1266
flax/nn/linear.py,sha256=-ojI1wtcMf95rC_yUX3gakfcYMH0Tw-su1RJ2Ndx0_I,13297
flax/nn/normalization.py,sha256=qDevLgKSVGXmNpgdm7Am0Kd4ZzkDtR8gjmMnGgMFRzg,9874
flax/nn/pooling.py,sha256=BuaMRBk17t0N_OeyPbWT9yTlIdkoAv2GiuYih2MXJFs,3353
flax/nn/recurrent.py,sha256=Dnj5MO7zBUhB3xDwMUhgFK-nwiMzkggRs-SVX5_NBlY,7407
flax/nn/stochastic.py,sha256=0o5f8A7lyn9B_i8-ZcYWpTDKpkfgdpqsg8Q7Xjwigqo,3445
flax/nn/utils.py,sha256=MPUZC-9xDNiZbAcfVA-iUf1n5QCM-4neS5PeD9TcoOQ,2492
flax/optim/__init__.py,sha256=3u9pXlP65GyrnG4vtERwIeR9aXxP7VfAEbm8HRUSUXs,1356
flax/optim/adagrad.py,sha256=PQMkMIrKcJN9BmW8W40x8OSmZbyb-i2bmyFRRQdeHdA,1858
flax/optim/adam.py,sha256=bxw4OiCJyFsDebXRnG1UUqwkaEX2NAYiCVE4NdyduOo,2833
flax/optim/base.py,sha256=Hh12P_h9VKpoNqg08FjvhcHgOr6kc4xFVtGaX9znJSo,18264
flax/optim/dynamic_scale.py,sha256=aLt_3vlNVLr937WRmSL_R_sUDBdTOFJWx1FqZUUOhgY,5424
flax/optim/lamb.py,sha256=gHkmy848N-5-JpJ8V6a09iByPTo4u1KDOYE13bRCXcQ,2980
flax/optim/lars.py,sha256=dixGdkcbwgNWIlTc1RuuDwMMt99j77qHjGJ4qDXsens,2963
flax/optim/momentum.py,sha256=XN3z_jphCm_mTOgPd6vvhxCssXg0avvwcywSYVUv33k,2255
flax/optim/rmsprop.py,sha256=RsE7Dp5-wZgqwO_HKZb2W1AzkhcpzaBLoIP71v3UfEw,2146
flax/optim/sgd.py,sha256=jtxc4qZONaQG-yb0Xz9xVXl3nWnrdH63PcScnQUQ8jA,1420
flax/optim/weight_norm.py,sha256=faf312-9i1f0fKI8qVy0lKN_KmqukOTXbXs0aCuoKJY,5936
flax/testing/__init__.py,sha256=ZRv8pGRJ52783jPvKkujUsyX5YAZ_pc8zgTgLt_0XCM,1250
flax/testing/benchmark.py,sha256=2X3UdPVEboFVnebjqFHl4qo4wjmuqwLtQ9_aF9mqGKA,9348
flax/training/__init__.py,sha256=VwFrZBz-oYzfgyy8UNpF9oyVjq2_9_gsgr7USeKLwh0,633
flax/training/checkpoints.py,sha256=me-9zjnE9pMsZVbWnojF7gMzdnfNNU7bhlvZjUBZYmY,5137
flax/training/common_utils.py,sha256=IcAWzxF6zw6nmvaplqj1jRrGnS0oV78ulfejwiIQMHo,1769
flax/training/lr_schedule.py,sha256=qKvboP_wbMISHmypa1_l90WAKfEsy1pp8eW4rSJ6Xyo,4980
flax/training/prefetch_iterator.py,sha256=W-LK4EgQNroI0a0wEVYIzleSCOlqXrOLkyOt3beu8Io,2761
flax-0.2.0.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.2.0.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.2.0.dist-info/METADATA,sha256=NMlObSyLHueX6UJ_l0bcNMdAhMKap2H7PdRKxIxTc-E,10583
flax-0.2.0.dist-info/WHEEL,sha256=g4nMs7d-Xl9-xC9XovUrsDHGXt-FT0E17Yqo92DEfvY,92
flax-0.2.0.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.2.0.dist-info/RECORD,,
