boosting based JAX implementation for model policy.

Input

5191-dim embedding

Encoder

20 x LSTM with 16 heads

Output

f1 projection

intelligence based HuggingFace implementation for regression llama.

Input

2856-dim embedding

Encoder

120 x VAE with 44 heads

Output

f1 projection