GPT Writes One Token at a Time. This Model Writes Entire Blocks at Once.

Building a Diffusion Language Model from Scratch with Flax NNX on TPU