Add device kwarg to Challenge for TPU/XLA support#268

kunal-mansukhani merged 3 commits into

mainAlphaGPU/leetgpu-challenges:mainfrom

device-kwargAlphaGPU/leetgpu-challenges:device-kwargCopy head branch name to clipboard