I've been putting off the GPU integration for too long. Help me add GPU support using wgpu, treating me like someone who doesn't have much experience with GPU programming in general.
I'll help you add GPU support to your ML framework using wgpu. Since you mentioned you don't have much GPU programming experience, I'll explain concepts as we go.
Before diving into code, let me explain a few key concepts: