Skip to content

JAX / XLA adding the importing external dlpack-aware Python arrays. #6

@adam-hartshorne

Description

@adam-hartshorne

Not a bug, but I thought I would make you aware of this pull request from the main JAX devs,

[JAX] Implement importing external dlpack-aware Python arrays, which allows for creating jax.Arrays from external GPU arrays
asynchronously.
jax-ml/jax#17238

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions