Skip to content

look if we can benefit from WebGPU support #575

Open
#1240
Open
look if we can benefit from WebGPU support#575
#1240
Labels
priority-1Needed for production usescope-discojsRelated to Disco.jstype-featureNew feature or request

Description

@martinjaggi
martinjaggi
opened on Apr 7, 2023

WebGPU just got released, and it might allow us to benefit from GPUs in the browser finally. TFJS actually also just started supporting it, at least with this new backend: https://www.npmjs.com/package/@tensorflow/tfjs-backend-webgpu

anyone interested to give this a spin, and see if it results in any ML training time speedups?

Activity

  1. martinjaggi
    added
    type-bugSomething isn't working
    type-featureNew feature or request
    and removed
    type-bugSomething isn't working
    on Apr 7, 2023
  2. s314cy
    self-assigned this
    on Apr 17, 2023
  3. hleepfl
    added
    scope-discojsRelated to Disco.js
    priority-1Needed for production use
    on Jul 15, 2026
  4. hleepfl
    unassigned
    s314cy
    on Jul 15, 2026

  5. JulienVig commented on Jul 17, 2026

    @JulienVig
    JulienVig
    on Jul 17, 2026
    Collaborator

    Burn also supports WebGPU. Supporting WebGPU is essential for LLM training.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    priority-1Needed for production usescope-discojsRelated to Disco.jstype-featureNew feature or request

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions