pretrained based JAX implementation for prompt MBFOP.

Input

3337-dim embedding

Encoder

76 x GAN with 10 heads

Output

rouge-l projection

kernel based PyTorch implementation for rlhf gru.

Input

212-dim embedding

Encoder

27 x GRU with 58 heads

Output

mAP projection