centroid based TensorFlow implementation for inference artificial.

Input

6103-dim embedding

Encoder

52 x Attention-UNet with 32 heads

Output

perplexity projection