transformers

53eade1f - Improve testing by adding torch_device to inputs and adding dependency on scatter