Lets make it possible to test WebGPU accelerated code via scripts, e.g. when running: https://github.com/PAIR-code/tiny-transformers/blob/main/animated-transformer/src/lib/seqtasks/tiny_worlds_train_self_att.script.ts
This may help us debug memory leaks too.
Breadcrumbs:
General TFJS platform guide: https://www.tensorflow.org/js/guide/platform_environment
WebGPU support for tfjs:
TFJS Node support (references GPU support too, but for linux and windows, not yet mac)
Key issues for Deno (looks resolved... uses tfjs-backend-webgpu and seems to use Chrome GPU backend support):
Possible alternative paths... webgpu for node and then make tfjs-node-gpu
Lets make it possible to test WebGPU accelerated code via scripts, e.g. when running: https://github.com/PAIR-code/tiny-transformers/blob/main/animated-transformer/src/lib/seqtasks/tiny_worlds_train_self_att.script.ts
This may help us debug memory leaks too.
Breadcrumbs:
General TFJS platform guide: https://www.tensorflow.org/js/guide/platform_environment
WebGPU support for tfjs:
TFJS Node support (references GPU support too, but for linux and windows, not yet mac)
Key issues for Deno (looks resolved... uses tfjs-backend-webgpu and seems to use Chrome GPU backend support):
Possible alternative paths... webgpu for node and then make tfjs-node-gpu