Skip to content

The _is_jax_zero_gradient_array check is no longer necessary #427

Open
Open
The _is_jax_zero_gradient_array check is no longer necessary#427
@NeilGirdhar

Description

@NeilGirdhar
NeilGirdhar
opened on May 14, 2026

Looks like Jax fixed their interface: jax-ml/jax#20620

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