Accelerating Capsule Networks with Tensor Comprehensions
Scalable Parallel Programming with CUDA