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

Open

Labels

backend compatibilitybeginner friendlyhelp wantedjax

opened

Description

jax mode is missing a dispatch for the NonZero Op.

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

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

Metadata

Assignees

No one assigned

Labels

backend compatibilitybeginner friendlyhelp wantedjax

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests