Skip to content

svd should support m < n on GPU #336

Open
Open
svd should support m < n on GPU#336
@sebffischer

Description

@sebffischer
sebffischer
opened on May 15, 2026

we need gpu argument in stablehlo lowering rules for that or rewrite the custom call in PJRT ...

Also, note the transpose trick via the layout that is used in JAX.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    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