Philippe Hansen-Estruch

icon indicating a website link https://philippe-eecs.github.io/website/

icon company UT Austin icon location Austin, TX icon location Interested in Generative Modeling, Reinforcement Learning, and Representation Learning

Results 1 issues of Philippe Hansen-Estruch

Error with putting arrays on CPU in cloud TPUs

1 comment

Hi I've been setting up big vision on a v4-32 TPU pod and I run into this error whenever I call `u.put_cpu` jaxlib.xla_extension.XlaRuntimeError: INVALID_ARGUMENT: Cannot copy array to non-addressable device...