clip-jax

View on PyPIReverse Dependencies (0)

0.0.5 clip_jax-0.0.5-py2.py3-none-any.whl

Wheel Details

Project: clip-jax
Version: 0.0.5
Filename: clip_jax-0.0.5-py2.py3-none-any.whl
Download: [link]
Size: 99230
MD5: 3d895f5cb2a752a57b3cf5a956dd13ce
SHA256: 464eeb9b41e3452b1cb51657dd39369dae0c0dfbee15780290005a6a77b29995
Uploaded: 2024-07-04 22:42:43 +0000

dist-info

METADATA

Metadata-Version: 2.3
Name: clip-jax
Version: 0.0.5
Summary: Training of CLIP in JAX
Requires-Dist: aqtp
Requires-Dist: flax
Requires-Dist: jax
Requires-Dist: numpy
Requires-Dist: orbax-checkpoint
Requires-Dist: tensorflow-io[tensorflow-cpu]
Requires-Dist: transformers
Requires-Dist: wandb
Requires-Dist: black[jupyter]; extra == "dev"
Requires-Dist: isort; extra == "dev"
Requires-Dist: optax; extra == "dev"
Requires-Dist: precondition-opt; extra == "dev"
Requires-Dist: tqdm; extra == "dev"
Provides-Extra: dev
Description-Content-Type: text/markdown
License-Expression: Apache-2.0
License-File: LICENSE.md
[Description omitted; length: 6722 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.25.0
Root-Is-Purelib: true
Tag: py2-none-any
Tag: py3-none-any

RECORD

Path Digest Size
clip_jax/__init__.py sha256=up3Wn1tooJUQlZZkOZogsmzEq_VNXrFbYzdqjjDNs64 122
clip_jax/data.py sha256=iVbp-VF45P-u65YaIi77szmRTBl0bLgJAwHbkPrttsM 16190
clip_jax/maxtext_model.ipynb sha256=dsyYy7mWzeMwh9V2cDWLe7Qki6-SkrY8aNmFqBjYQ4Q 16555
clip_jax/modeling.py sha256=rQ3_TS6Mr-ib5z4U71EXMpcTvycWdGQZIZg4ulZH2d8 78357
clip_jax/partitions.py sha256=cO1cjjHsBccuiG2RG4RJD4P8O6ceyAR2JLb4GuQFPHg 3072
clip_jax/tokenizer.py sha256=PvkbAoDVKuxh9BnKBh1FSvgHpF4fsM7Cdo2Ewf0ebcA 135
clip_jax/utils.py sha256=60U-9bxp_0RWcA_slqM5onK3UnBcOMC_p5g1Wa4yXIU 1180
clip_jax/wandb_utils.py sha256=nzOPv7HN6NBRFCoGFSz-t82siNOcr9-xBtcEfRReA_4 1904
clip_jax/maxtext/README.md sha256=ToYcaS8tDEz4Ee_5o1f4T3cDznJd70Ke7D5JMzPGyTo 74
clip_jax/maxtext/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
clip_jax/maxtext/checkpointing.py sha256=4dTCMcipLli1R2x39pLngSPQO3jrb_t2bR9oJkJBg1Y 8394
clip_jax/maxtext/common_types.py sha256=Z7HBw7ROtYcX60CSSazXO6Br-uKJE-N1m74O4fx2GO4 1144
clip_jax/maxtext/inference_utils.py sha256=rToYZEBLcSs_aCLXgU1cybinm_pTQPG5W3I99Xn5Mj0 3259
clip_jax/maxtext/max_logging.py sha256=QPDSETlQkYOeuN11P5OasfiwPVqsvVWF87PmR1fcXOc 700
clip_jax/maxtext/max_utils.py sha256=dWQB9zcaQy4dFU827Q3T-1II6nbWEWYMry6nJpA0nGc 24841
clip_jax/maxtext/multihost_dataloading.py sha256=EDaJy2hgdx79ZNmHlHySIFKmUPlUz2Kg8WVDOb73ko4 4403
clip_jax/maxtext/sequence_packing.py sha256=7LEHpjY0G-ay50GxI7VjjXWcs11Nez-Zgx807U-_wS0 7859
clip_jax/maxtext/tokenizer.py sha256=PUCkqeC7jh3JmFXYFzsvV9PwkR357dgcxAKON1lKeRs 2065
clip_jax/maxtext/input_pipeline/_grain_data_processing.py sha256=CFz5_LF0UgqI12INv2D5qE252iuQENUBdhR8r8D410k 5356
clip_jax/maxtext/input_pipeline/_grain_operations.py sha256=pYz0pY-WPwQBSqDNfpXNJlhmPZSYdpwcs-01cNJkvy8 3781
clip_jax/maxtext/input_pipeline/_grain_tokenizer.py sha256=GyHsmY7lNBMIj5LFraWmIEiNtPQlaIAVThXzgkjVchU 2640
clip_jax/maxtext/input_pipeline/_tfds_data_processing.py sha256=xALA2yenDcbBab4XvcfmKr3y3GRW5bSEHfiazDt-Jms 8319
clip_jax/maxtext/input_pipeline/_tfds_data_processing_c4_mlperf.py sha256=00z9mt4YtJg3eaHi2dF4ic2pLtvPQ-VPIXZeOwTpbrg 11338
clip_jax/maxtext/input_pipeline/input_pipeline_interface.py sha256=1QrkBz6OLuIZptuiuBpGNK6LG8CYoyV4q2IttJHp25A 6318
clip_jax/maxtext/layers/attentions.py sha256=t9vF9i5z_KMNFOTI5onf3UGuIhT3u4nMomf1hWGX7FI 42524
clip_jax/maxtext/layers/embeddings.py sha256=pU-FDl9sqfxjvZbkRUG0TvlKi0TEC_2x5iOIhm8dJKU 7564
clip_jax/maxtext/layers/gemma.py sha256=vTNl0Y10MEJr_xWq1LdDBrzr0iMyMQMHX0FoMlzt9g4 5094
clip_jax/maxtext/layers/gpt3.py sha256=-jI2wlFSwbUieZgeC7fRja2ieRCeRK_tHaSmCX31emw 12364
clip_jax/maxtext/layers/initializers.py sha256=gozYT1kFwVvzdmnUrSYc99GZUeGdim2aRNwm0wfMzV0 1708
clip_jax/maxtext/layers/linears.py sha256=Z2DXt7VayJwdWuwRbsUGbju_9o09j_8P68xn4mvQFwA 11967
clip_jax/maxtext/layers/llama2.py sha256=5djSfQFTn7II7IkJFR64AezeJWlLHRtpsu2UFwUhGgQ 5293
clip_jax/maxtext/layers/mistral.py sha256=FTWjgp2sjlIJiuUaD4wOepVTFX009mY1GBDVxrKAFs0 7604
clip_jax/maxtext/layers/models.py sha256=C4pChMONVQqhSybzhp0FNNSIQlL5UrnX4-SnLpxOZSg 17030
clip_jax/maxtext/layers/normalizations.py sha256=d-vjXYrMlDm3mjPUB-8EbNMT4Juq6UazTZJ7P3pYz20 1626
clip_jax/maxtext/layers/quantizations.py sha256=f6OZowBbalrV2D_txL9zOjaa8ZoMfOE_KAyW1NwFp_Q 5103
clip_jax-0.0.5.dist-info/METADATA sha256=3S4iBwvmHyrtgxiLVWLHtmqtyzGQYNcCn6keeqHQhQQ 7342
clip_jax-0.0.5.dist-info/WHEEL sha256=fl6v0VwpzfGBVsGtkAkhILUlJxROXbA3HvRL6Fe3140 105
clip_jax-0.0.5.dist-info/licenses/LICENSE.md sha256=NHdnngp-AU4ebNqmqwtQC4hcLpamPC1iCwav_fYK46Y 11354
clip_jax-0.0.5.dist-info/RECORD