Add CLIP-JAX ADE20K implementation, training, and tests

[Thakor-Yashpal]

Resolves #<issue_number_goes_here>

Reference

Checklist

• [] I have read the Contribution Guidelines and used pre-commit hooks to format this commit.
• [] I have added all the necessary unit tests for my change. (run_model.py for model usage, test_outputs.py and/or model_validation_colab.ipynb for quality).
• [] (If using an LLM) I have carefully reviewed and removed all superfluous comments or unneeded, commented-out code. Only necessary and functional code remains.
• [] I have signed the Contributor License Agreement (CLA).

[jenriver]

Hello Thakor, it'd be really cool to have a CLIP implementation here. It seems there's no file changes here, did you forget to push commit?

Related issue: #72

[Thakor-Yashpal]

Hello Thakor, it'd be really cool to have a CLIP implementation here. It seems there's no file changes here, did you forget to push commit?

Relaetd issue: #72

I had pushed the initial code earlier, but the implementation wasn’t up to the quality I wanted, so I removed those files. I’m making the necessary improvements now and will push the updated files on Wednesday.

[Iamleos]

When CLIP will be merged?

[jenriver]

Can this be entirely used in tests/ insetad to follow pattern in https://github.com/jax-ml/bonsai/blob/main/CONTRIBUTING.md#contributing-a-model?