* test unet blocks 2d * change to randn_tensor * mps flaky
Attach a file by drag & drop or click to upload
