├── .gitignore ├── CLIP.png ├── LICENSE ├── MANIFEST.in ├── README.md ├── clip_jax ├── __init__.py ├── bpe_simple_vocab_16e6.txt.gz ├── clip.py ├── model.py └── simple_tokenizer.py ├── model-card.md ├── notebooks ├── Interacting_with_CLIP.ipynb └── Prompt_Engineering_for_ImageNet.ipynb ├── requirements.txt ├── setup.py ├── tests ├── bench_tpu.py └── test_consistency.py └── tpu_install.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/.gitignore -------------------------------------------------------------------------------- /CLIP.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/CLIP.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | include clip_jax/bpe_simple_vocab_16e6.txt.gz 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/README.md -------------------------------------------------------------------------------- /clip_jax/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /clip_jax/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/clip_jax/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /clip_jax/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/clip_jax/clip.py -------------------------------------------------------------------------------- /clip_jax/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/clip_jax/model.py -------------------------------------------------------------------------------- /clip_jax/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/clip_jax/simple_tokenizer.py -------------------------------------------------------------------------------- /model-card.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/model-card.md -------------------------------------------------------------------------------- /notebooks/Interacting_with_CLIP.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/notebooks/Interacting_with_CLIP.ipynb -------------------------------------------------------------------------------- /notebooks/Prompt_Engineering_for_ImageNet.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/notebooks/Prompt_Engineering_for_ImageNet.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/setup.py -------------------------------------------------------------------------------- /tests/bench_tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/tests/bench_tpu.py -------------------------------------------------------------------------------- /tests/test_consistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/tests/test_consistency.py -------------------------------------------------------------------------------- /tpu_install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kingoflolz/CLIP_JAX/HEAD/tpu_install.sh --------------------------------------------------------------------------------