Add subgroups feature support

[beaufortfrancois]

Feature request

As reshared by @FL33TW00D at https://x.com/fleetwood___/status/1894754562210165029, having subgroups support in transformers.js would be huge for performance.

I'm filing this feature request to engage conversation and discuss how this can be achieved now that WebGPU subgroups have shipped in Chrome 134: https://developer.chrome.com/blog/new-in-webgpu-134#improve_machine-learning_workloads_with_subgroups

Note that some work has been started in Apache TVM as well in apache/tvm#17699

Motivation

Performance, performance, and performance.

Your contribution

I'd be happy to help answering questions about how subgroups are implemented in Chromium.

[xenova]

Exciting! 🚀 Let me loop in @guschmue to the discussion to see where we can add support for this 💪