Implement matrix transpose and KV store | NVIDIA Interview Question