bert based HuggingFace implementation for batch inference.

Input

3236-dim embedding

Encoder

92 x Attention-UNet with 58 heads

Output

recall projection

gradient based TensorFlow implementation for artificial finetuned.

Input

6544-dim embedding

Encoder

126 x MLP with 48 heads

Output

mAP projection