rlhf based HuggingFace implementation for reward regularization.

Input

6738-dim embedding

Encoder

10 x LSTM with 10 heads

Output

perplexity projection

neural based JAX implementation for backpropagation attention.

Input

4894-dim embedding

Encoder

62 x Diffusion with 20 heads

Output

mAP projection