Wheelodex » Projects » jax-models

jax-models

View on PyPI — Reverse Dependencies (0)

0.0.5 jax_models-0.0.5-py3-none-any.whl

Wheel Details

Project: jax-models
Version: 0.0.5
Filename: jax_models-0.0.5-py3-none-any.whl
Download: [link]
Size: 46120
MD5: d077a94f6fee006cd5378ad7d737a507
SHA256: 42de33bbd3cb2dbf24dfbf662af9a0f8c0b6b38cf0b20df20d365bf639b529d1
Uploaded: 2022-02-27 18:32:18 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: jax-models
Version: 0.0.5
Summary: Unofficial JAX implementations of deep learning research papers
Author: Darshan Deshpande
Author-Email: darshan.g.deshpande[at]gmail.com
Home-Page: https://github.com/darshandeshpande/jax-models
Keywords: jax flax models computer-vision deep-learning transformers
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3.6
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.6
Requires-Dist: flax (>=0.3.6)
Requires-Dist: tqdm
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 5935 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.37.1)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
jax_models/__init__.py sha256=qI0iDMjhe4aKeag5uqyEzfvwiyDf2jbzKEkYO3iZr54 159
jax_models/activations/__init__.py sha256=vudv6KGGyfoca624tzymfZ4dWIfSHdLMSganpwn5Ibg 55
jax_models/activations/activations.py sha256=xLg99XwB3sHB9k6LIviTnAPwS2CmxxKguEOwLhUALXU 814
jax_models/initializers/__init__.py sha256=-TRoEkw8rrkmQyWVSrA3KopcSCJg-UlMtaakjSzLd8A 42
jax_models/initializers/initializers.py sha256=O8QTjsSELKZfdFufW9FxfN_Z8sbHuk4QvZUf_kxEmVg 199
jax_models/layers/__init__.py sha256=MEp9WMFRTamVy-4mHrDCV7G6VCSXTx0v7vlWidpLbWg 409
jax_models/layers/attention.py sha256=K2bJjq8h798NMzn6ViKXj_Le2CNTCVSEbgOlm4KslQ4 1381
jax_models/layers/depthwise_separable_conv.py sha256=j5tDzjXavs0-xY1s4Vglg5OLsSo4lY2Xb--wNTTndhc 2680
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=saGDE4VdigtAkWbR_cHw6jcKD_7c9vkd3esu3tF9WE4 1420
jax_models/layers/patch_util.py sha256=8svlkFqdbEM3WmyH_w5EQozwO-K2rhBjao7Es6OMBH0 2384
jax_models/layers/pool.py sha256=wuYCq0BbJTxODYx6x2W2gbF21uVyJRy-y-SFWXlpdwE 1136
jax_models/layers/squeeze_and_excite_layer.py sha256=cwLiOVmZch0hG55wBHROPPv-fmd2yxao3RmzC8dHllk 696
jax_models/models/__init__.py sha256=asW0tUDmIYcCnQsJ_fdyxlGF4RoapoTXWrrZ3dFR3VE 355
jax_models/models/cait.py sha256=yM21SnWfQHLZW0bz_sAzpGSuajR7LizWAQVNsHLaNRo 17822
jax_models/models/conv_mixer.py sha256=obQXeBhAED02wmIaruiFM3hfu3TJrlG7yOJvTMqwjP4 3981
jax_models/models/convnext.py sha256=5z--r4jIN_-scrWUhg7KHh3sm0faiPQpjFM5NpLVNGc 17466
jax_models/models/helper.py sha256=MHjFOgvc_XDlkFSOwCiZLJ9VlYg5S8zN-615RDcugBU 1400
jax_models/models/masked_autoencoder.py sha256=g8G0PZcluCcBjyJrJUzQN9TZmYc1eXb0osbFm09j3AQ 10224
jax_models/models/mlp_mixer.py sha256=uq7rO3BO_bMYJF_8xFXG06YHReXyESjKDwfXr6d5XyM 5628
jax_models/models/model_registry.py sha256=H9NmCpid9TjpkkWnirW2H-CAkFN_mvZAz012_SxzN2w 1955
jax_models/models/mpvit.py sha256=hS6DorNQaZY8DwFbcku2NL6u8LdzUkazRxufMnW601U 16040
jax_models/models/patchconvnet.py sha256=jHNwAtOPgKU8owipWUX-F3zeIgulZbzhcg2WSM5rg-s 9816
jax_models/models/poolformer.py sha256=luxwSUkPDtB3WXZaEf3eC6F7VhqFywEloKn8siDk1jQ 12006
jax_models/models/pvit.py sha256=0pxCM4x0gkSKH9zCRlKO83DM1q503_we4I1hWyuWnFc 13436
jax_models/models/segformer.py sha256=0zvDFvxHFf5zPOFVAfVQWxaKgezr-gx5e23YRfA0x-U 15213
jax_models/models/swin_transformer.py sha256=A7Xz4nfkl9YFpqyHg5VP3eVav-ITFFsHRPGT739d5OU 19596
jax_models/models/van.py sha256=qi0V0PPZPfnY2NP-oax-90IcHULb1aaZlm17S9cxwnI 10393
jax_models-0.0.5.dist-info/LICENSE sha256=F8KdRp48eBGKyes59FcfUoqCfYZ15cmHpL7gp760x_Y 11515
jax_models-0.0.5.dist-info/METADATA sha256=JvBrdTx8a-q7hAa6HWLSMwbsIZyqFqG4pucoVIXTtpE 7116
jax_models-0.0.5.dist-info/WHEEL sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA 92
jax_models-0.0.5.dist-info/top_level.txt sha256=cTqzsx7qgt4gVLo9BGf2JADv7toYhA1cLXqBtXAUTpw 11
jax_models-0.0.5.dist-info/RECORD — —

top_level.txt

jax_models