WebGPU for Compute: WGSL data type support

[nevi-me]

I'm interested in using WebGPU for compute, I specifically want to try an Apache Arrow compute backend.
I was going through the WGSL reference last weekend, and noticed that data type support is very limited.

If I wanted to create a kernel that say takes a Vec of u16 values, is this something that's not possible with WGSL, or am I missing something?

I would want to support most of the Arrow supported data types (unsigned, signed, float) as a start, and from the reference I can't see how I'd do that.

Is this something that'd be possible with WGSL as the shader language, or would I have to use some other shader language and use naga to transpile (if that even makes sense)?

Thanks