Avoid use of deprecated device_buffer attriutes of jax.Array#325
Merged
copybara-service[bot] merged 1 commit intomasterfrom test_589237893Dec 9, 2023
+1-1
Commits
Commits on Dec 9, 2023
authored andJake VanderPlascommittedChexDev