Add support for JinaAI Re-Rankers V1

[alvarobartt]

What does this PR do?

This PR adds the JinaBertClassificationHead in order to support the JinaBERT re-ranking models as e.g. jinaai/jina-reranker-v1-turbo-en. Note that JinaBERT's classification head is exactly the same as standard BERT's.

Warning

This PR is not yet ready to be merged, as there seem to be some discrepancies between the Sentence Transformers and ONNX outputs (those are aligned), but the outputs for the default Candle backend is not, meaning that there may be something off somewhere. Also it has yet only been tested on CPU and MPS devices, CUDA support is still missing / pending.

In order to test the enabled models, please refer the the following revisions (until those are merged):

• https://huggingface.co/jinaai/jina-reranker-v1-turbo-en/discussions/13

cargo run --release --features candle,http --no-default-features -- --model-id jinaai/jina-reranker-v1-turbo-en --revision refs/pr/13 --dtype float32

• https://huggingface.co/jinaai/jina-reranker-v1-tiny-en/discussions/11

cargo run --release --features candle,http --no-default-features -- --model-id jinaai/jina-reranker-v1-tiny-en --revision refs/pr/11 --dtype float32

Fixes #579

Before submitting

• This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
• Did you read the contributor guideline, Pull Request section?
• Was this discussed/approved via a GitHub issue or the forum? Please add a link to it if that's the case.
• Did you make sure to update the documentation with your changes? Here are the documentation guidelines, and here are tips on formatting docstrings.
• Did you write any new necessary tests?

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

[CoolFish88]

Hello @Narsil,

I managed to test the proposed changes in this PR and they solve the previous limitations.
I was wondering if you have some bandwidth to review this PR such that the solution can be merged into the main branch.

Thanks for the continued support in developing and maintaining this service!

[Narsil]

LGTM