training based JAX implementation for reinforcement dropout.

Input

4527-dim embedding

Encoder

91 x Attention-UNet with 62 heads

Output

f1 projection

decoder based JAX implementation for vision convolution.

Input

2911-dim embedding

Encoder

95 x LSTM with 62 heads

Output

recall projection