30 ms·
Question. I can see in the code the WGSL that's needed to implement inference on the GPU. https://github.com/0hq/WebGPT/blob/main/kernels.js https://github.com/
by ianpurton 4y ago
Question. I can see in the code the WGSL that's needed to implement inference on the GPU. https://github.com/0hq/WebGPT/blob/main/kernels.js https://github.com/0hq/WebGPT/blob/main/kernels.js
Could this code also be used to train models or only for inference?
What I'm getting at, is could I take the WGSL and using rust wgpu create a mini ChatGPT that runs on all GPU's?
- HopenHeyHi 4y agoNo, that is not how that works, sadly you cannot.
- FL33TW00D 4y agoWhy not? You could absolutely write the training code in WGSL.
- spencerchubb 4y agoThis repo only does inference but it should be possible to write training code that runs on WebGPU.