Skip to content

Add error message to NonZero Op telling users it cannot be jitted in JAX #1509

New issue
New issue
Open
Open
Add error message to NonZero Op telling users it cannot be jitted in JAX#1509
Labels
backend compatibilitybeginner friendlyhelp wantedExtra attention is neededjax
@jessegrabowski

Description

@jessegrabowski
jessegrabowski
opened on Jul 1, 2025

Description

jax mode is missing a dispatch for the NonZero Op.

There's a jnp.nonzero, so it should be easy to do.

Metadata

Metadata

Assignees

No one assigned

    Labels

    backend compatibilitybeginner friendlyhelp wantedExtra attention is neededjax

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions