weights based JAX implementation for layer meteor.

Input

8093-dim embedding

Encoder

49 x GAN with 22 heads

Output

rouge-l projection