I don’t know why, but you need to import the WebGPU extensions:
import '@babylonjs/core/Engines/WebGPU/Extensions/'
If you don’t do it, the extensions attached to the WebGPUEngine prototype are the WebGL ones…
cc @RaananW as it does not feel right.