Skip to content

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

@iislucas

Description

@iislucas

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions