Alex Birch
apply device-transfer patch from https://github.com/mosaicml/llm-foundry/pull/225/files
ec8bed8 unverified