Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support running scripts with Nodejs or Deno, including webgpu support #62

Open
iislucas opened this issue Feb 19, 2025 · 0 comments
Open

Comments

@iislucas
Copy link
Collaborator

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant