Skip to content

jax.device_put_replicated is deprecated, use jax.device_put instead #671

Open
Open
jax.device_put_replicated is deprecated, use jax.device_put instead#671
@zhuyeqing

Description

@zhuyeqing
zhuyeqing
opened on May 8, 2026

when i use brax 0.14 and jax 0.10, i will get error infomation like this : jax.device_put_replicated is deprecated, use jax.device_put instead

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