Just image Transformer.

Track a series of values and provide access to smoothed values over a window or the global series average.

Image to Patch Embedding

The final layer of JiT.

Embeds class labels into vector representations. Also handles label dropout for classifier-free guidance.

Embeds scalar timesteps into vector representations.