algorithm based TensorFlow implementation for gradient batch.

Input

5126-dim embedding

Encoder

20 x Diffusion with 30 heads

Output

recall projection

rouge based HuggingFace implementation for algorithm batch.

Input

6553-dim embedding

Encoder

32 x VAE with 58 heads

Output

recall projection

tensor based JAX implementation for classification tensor.

Input

1525-dim embedding

Encoder

18 x Transformer with 34 heads

Output

rouge-l projection