8000 JAX as a possible dask array backend? · Issue #11983 · dask/dask · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
JAX as a possible dask array backend? #11983
New issue
New issue
Open
Open
JAX as a possible dask array backend?#11983
Labels
needs triageNeeds a response from a contributor
@dpanici

Description

@dpanici
dpanici
opened on Jun 11, 2025

Hi,

I am interested in using dask as a vehicle for parallelization of an existing CPU/GPU code which is written using JAX. I am pretty naive/new to dask, but I see that there is some documentation on use for GPUs/ a blog post where the backend for array creation can be cupy, for example. Is there something similar planned/available for JAX?

Thanks!

Metadata

Metadata

Assignees

No one assigned

    Labels

    needs triageNeeds a response from a contributor

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions
      0