Show HN: OpenCLIP Jax – Implementation of OpenAI's Clip in Jax/Flax (github.com)

open_clip_jax is an open source JAX/Flax implementation of OpenAI's CLIP, including image and text towers, pre-trained parameters, training utilities that can scale to TPU Pods, and more. It is inspired by but not affiliated with OpenCLIP and aims to deliver similar functionalities with a JAX backend.

Questions and feedback are welcome in the comments sections.