jax_models/__init__.py,sha256=zdM3gWtDTpNbvr7qKRWMwU74sJZdUOs6KJTQsfwEwrg,71
jax_models/activations/__init__.py,sha256=vudv6KGGyfoca624tzymfZ4dWIfSHdLMSganpwn5Ibg,55
jax_models/activations/activations.py,sha256=xLg99XwB3sHB9k6LIviTnAPwS2CmxxKguEOwLhUALXU,814
jax_models/layers/__init__.py,sha256=J9REMPEP6zl28Qu4AgHxNeE7MDoDbV8xTj9dADiw7ws,367
jax_models/layers/attention.py,sha256=K2bJjq8h798NMzn6ViKXj_Le2CNTCVSEbgOlm4KslQ4,1381
jax_models/layers/depthwise_separable_conv.py,sha256=crjVA2fejA3ccobv7xUexcZ9OQ8uuMtMUw7E6rmc5yE,2681
jax_models/layers/drop.py,sha256=WSwobiUN0kpHtcc3bkqhal1Iggz_z3_IuNWqbiMS5SA,917
jax_models/layers/mask.py,sha256=PLUeiczxIDOVgPuM_h2ycyt5aVJwFAFSuDgZZtM-peU,834
jax_models/layers/mlp.py,sha256=kyHJiJkD3Iyx6knUo0bY20OdWr7se-r6BuEZgIjXUqE,1138
jax_models/layers/patch_util.py,sha256=GPgHDSe6FgGUM39yjaUwG2I4wA2C9DzPamVTarO4vQg,1490
jax_models/layers/pool.py,sha256=EO7N27FBPn31GhyNiT-LMVeDO2rU2EbLNktf1yJSaro,524
jax_models/layers/squeeze_and_excite_layer.py,sha256=cwLiOVmZch0hG55wBHROPPv-fmd2yxao3RmzC8dHllk,696
jax_models/models/__init__.py,sha256=-jVgSuP7_RwjlkIyiqjVTCf0_XhYPAvg5SjQkN1ynck,238
jax_models/models/conv_mixer.py,sha256=OlDUXFET4wIXJC3Ofab-uKObNuNGYMG2aH2X3taf-Q8,2272
jax_models/models/convnext.py,sha256=A1qqHDJeU5rQ8d8uQW57Lvl1D2lVE5zH1xZZo700MBA,4508
jax_models/models/masked_autoencoder.py,sha256=gz7mfDKMJZVU895mLqch0oMK7yIc_FavOvh945Qo2Nc,8604
jax_models/models/mlp_mixer.py,sha256=7vbRmsH4LZM_V5X9ovbFRZJGgnVq5kY4JK68r4GHgHY,3344
jax_models/models/model_registry.py,sha256=cDEzsxmck9oX8yrHYbkkt05EpP02k2WpEo2jV79ZE_s,2208
jax_models/models/mpvit.py,sha256=f0utviOe1s-vVsu3bcIlbNgMzaW-9_RC1Tft0-kQHkw,14144
jax_models/models/patchconvnet.py,sha256=sIUT1n_KqOXcmoLaNU6HbzzZK-uCd3rlMUwY8V9g0iY,7570
jax_models/models/poolformer.py,sha256=mx0zK-E0u6GHqEvTfnJEttkOOuXdYCKJIrYBJYCYOmk,8885
jax_models/models/segformer.py,sha256=KnMr4wfTgHPbu9AWJ2BzPgto5IMGsvaOVHIFtQD2naQ,12800
jax_models/models/swin_transformer.py,sha256=VNtxCkoZPlXbxOrqUNXJ4Hlwg3vLeX0ZLoJ2HG5Xfgc,14890
jax_models-0.0.3.dist-info/LICENSE,sha256=F8KdRp48eBGKyes59FcfUoqCfYZ15cmHpL7gp760x_Y,11515
jax_models-0.0.3.dist-info/METADATA,sha256=WwQlmwK2g3dKe7v55byvekPbX0J0L_jCqJfcxk1K24s,6623
jax_models-0.0.3.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
jax_models-0.0.3.dist-info/top_level.txt,sha256=cTqzsx7qgt4gVLo9BGf2JADv7toYhA1cLXqBtXAUTpw,11
jax_models-0.0.3.dist-info/RECORD,,
