The array API standard will soon require the `device` keyword be added to `astype` See: https://github.com/data-apis/array-api/pull/665