Should be quick and easy with WebGPU, too.

That's an even better idea, I bet this could run in Transformers.js.

Good idea. Could you make that.