Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Detection of Jax Arrays Breaks on Jax=0.4.* #2535

New issue
New issue
Open
Open
Detection of Jax Arrays Breaks on Jax=0.4.*#2535
Labels
area / integrationsIssue area: integrations with other tools and libshelp wantedExtra attention is neededtype / bugIssue type: something isn't working
Milestone
v3.18.0
@EntilZha

Description

@EntilZha
EntilZha
opened on Feb 7, 2023

The detection for jax arrays here

aim/aim/sdk/num_utils.py

Line 66 in 285a09f

def is_jax_device_array(inst):
is broken in Jax 0.4.* since DeviceArray was renamed to Array. The simple fix to support the prior/current version would be to check for either DeviceArray or Array

Metadata

Metadata

Assignees

No one assigned

    Labels

    area / integrationsIssue area: integrations with other tools and libshelp wantedExtra attention is neededtype / bugIssue type: something isn't working

    Type

    No type

    Projects

    Aim roadmap

    Status

    Next-up

    Milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions