I think Bun (which has typescript embedded but written in zig) would be a great place to experiment.

I could see it adding some tensor types that drop into zig code that uses webgpu bindings. And add some operator overloading so you could just add t1 + t2 etc.