Add FlaxCLIP
# What does this PR do?

This PR adds the CLIP model in JAX/Flax.
- Added jitted tests for `get_image_features` and `get_text_features`
- `__init__(...)` now takes `input_shape` as an arg, when `None` it's set using default values in `config`.

Merging!