google / google/brax

jax.device_put_replicated is deprecated, use jax.device_put instead

Open
#671 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
3.2k
Forks
349
No merged PRs in 30d

Description

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

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.