flax/__init__.py,sha256=IS1U8vdWd1mwkOdNoewfA-BPkl4AIENfNlbZkWisnPw,861
flax/config.py,sha256=p-HJ6EpVqkFBokshcs_9qAZUjaNED3ZZGWAKKOv2D-I,2104
flax/errors.py,sha256=BTPU5fxTQd8t6sK4GaKLHv0reKXYhmw8tIaH_ZESNU8,19525
flax/jax_utils.py,sha256=VZoRnhbI8l-QCrxDV-Lp6Hmdx7KscksrsbJqiI8CNKc,8775
flax/py.typed,sha256=Tax5VCwEb0DFaIMIb4fcDTU0qMhkF1jfe-0zVWwzftE,58
flax/serialization.py,sha256=t-tgofomeLop2Ien_-sVk4-Fe_5Yvle1fLtluY9GzcU,12733
flax/struct.py,sha256=nzKeJHpwBoYj6LaDoTKZtZgUz-wavnHafGfNF8wV6vo,7376
flax/traceback_util.py,sha256=ozGNM-0mlncdvfQKXnqCXIcYpXzt0LLDs9tXUkGBnoY,1989
flax/traverse_util.py,sha256=iGk2s_UJ3QqXZNfrxRDRM9yIJ_M-qYU--gLHCmyQYWc,11306
flax/version.py,sha256=YLWSeq_bzWVHNQJk8mGkc-dDvLk-IC0XXA2yAoj1KHg,605
flax/core/__init__.py,sha256=e25QtojOhbSCqyQiKcdbd8AToxavCsRWdy74NamSJ04,831
flax/core/axes_scan.py,sha256=QU2wYcIMuCZGOGUmZkETCNrQorckxNDcGpRTZqLgOI8,5045
flax/core/frozen_dict.py,sha256=CXku9J5SbEKFpAjxmIJOR8vz0uTrNWhOP6JMRj29Lts,6005
flax/core/lift.py,sha256=b-Idndx_EXvnr8yHM6mu6STrECd_3kzhkYjtxko4xEg,44114
flax/core/scope.py,sha256=goBFGxXZ64JSTQvAkduWh3EN_RU-ox1f8hiVKlvj5JU,28572
flax/core/tracers.py,sha256=xzAymR9fu0fPENJcQ7y527zsQmPaed2SWyH2T9nKT8A,1054
flax/core/variables.py,sha256=AVLWdjdHA9jVE0p6ny_gYpG1W3FcAvFulqqGhkTUII0,1558
flax/core/nn/__init__.py,sha256=nw4Td51jARhfNHUw0B4aQLVSG2qjdqZmxX1CGgP836s,1378
flax/core/nn/attention.py,sha256=Gfv8YMB1gwGSo5MnUTrobYXMshaV21chjbx2j8HBHjw,18495
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=cNm1c0B7OK4CfR2uXhH37FYR22bcpPb3ROR1SbsdAx0,1814
flax/linen/activation.py,sha256=_NZTPtXlJh6oYp93ORR-5jfoBwPA3p2ibcsFGiElTNw,2063
flax/linen/attention.py,sha256=iiXVmKYI8KhdDx4f3O5P5cGTKMX1sFJVuUz65PbsGGg,17791
flax/linen/dotgetter.py,sha256=K0oNBdvXOiJX38keqpEuPc_UEDlCkk7UCsDW0qzGsAo,3106
flax/linen/initializers.py,sha256=nMFH9LwfxGPs3F1cbmuxQHtNFZD1vj6h5dI_L0oL-aM,1474
flax/linen/linear.py,sha256=EOVg-hLfHBuR4nY9vgifC0X2mHcDySwBOiNQfEOj83w,19796
flax/linen/module.py,sha256=HEstRo8tyyUmMKBlAW1unjnGXVou6cHkiyc53R8QOec,55818
flax/linen/normalization.py,sha256=8Ec509Sw-7jiiu_iFMjUTj4a9srTKsZ6GbVsTmDCNhY,14838
flax/linen/partitioning.py,sha256=KeDOIo4bUseFEIiMS5XEnQknrEW5vVoMWcq7sdrYScw,20921
flax/linen/pooling.py,sha256=Lmxot95hub5FjPsWJDvgj3owCdDfIUzLsAVYf_zkc4A,4926
flax/linen/recurrent.py,sha256=SA2Gj5J5fqld59GD1XfKZeZNwn23HG0aEEEGm9Ncy38,18397
flax/linen/stochastic.py,sha256=1HegqFmmB_Ae1STkkLUjEve4RDzeZXmVYvhLl6v4W3I,2257
flax/linen/transforms.py,sha256=b9YjBDjXARdJ45wrkIdgdqDlYOF1p1oaS2Y8_phTD-0,42199
flax/metrics/__init__.py,sha256=YtkS-Bp6L4D50sa6m-MGVRb3ReK_APAL0UmEYY2yWLw,582
flax/metrics/tensorboard.py,sha256=hd4fQE3KHpC_vuV2S5Ir-GgvyZ68oge7cgcrYs3NXmY,6797
flax/optim/__init__.py,sha256=6Lf8B6eKP_ft7HFSBiwoAT7qBbGDQjYNajHtGYWmnzI,1902
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=04Xb_ycYWPPcCYwYy9iZz7kMTRE0ISQznxOp_2enY0Y,20327
flax/optim/dynamic_scale.py,sha256=g1tu9-Bwek7fIlh0GGW3ROvSbfDgtTO4MEE6TVMdrrQ,5386
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/testing/__init__.py,sha256=hKWEbKmPH84atZdxK30BEo7K4kK7IqLrCizoC_s3Uiw,667
flax/testing/benchmark.py,sha256=1CSZQ6vf-KLCmhrHYcxpKrakfcaWQw9hPVbGdiVnA_A,9562
flax/training/__init__.py,sha256=TE21K5jkA5ZP28dHY-CtjfSeVMb4i3a4XUw2GynSnNc,633
flax/training/checkpoints.py,sha256=f3BThqhuo2msrwEXHcb5h-dHx8G9Es1q5pg1CwLp2to,12710
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-0.4.0.dist-info/AUTHORS,sha256=xAt9eymRM6GSWVKw3mzCuF2jJRPeNqSwWMFJshxt-aE,293
flax-0.4.0.dist-info/LICENSE,sha256=Jeh5HaJvfnTEuvGer9fRN7ZKRpUu-EDtAcFytoDPQEY,11309
flax-0.4.0.dist-info/METADATA,sha256=Gxp-VUjHUwKfMsY3E8Y2XN17yMBW566nqq-jqJjMT3o,9171
flax-0.4.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
flax-0.4.0.dist-info/top_level.txt,sha256=r_wO1qpXX1_wNUJ0BDDhQQ9F-2yi8Igav-p2VTwO_tM,5
flax-0.4.0.dist-info/RECORD,,
