Add `DeviceTensorTrait` to export annotations #207

Job	Run time
Unit Tests and Type Checking (3.11, ubuntu-latest)	1m 37s
Unit Tests and Type Checking (3.11, nodai-amdgpu-mi300-x86-64)	1m 36s
	3m 13s

Provide feedback