I would be happy to join this project. I’m super excited about JAX and have a little experience with it. My main background lies in computer Vision with Tensorflow and i have some experience with CLIP-like architectures and zero-shot classification.
Hope I can add something to the project!