/clip-jaxtorch

A simple JAX implementation of CLIP using jaxtorch

Primary LanguagePython

Relatively simple JAX implementation of CLIP

Usage:

model, params = clip_jaxtorch.clip.load('ViT-B/32')