nest is a lightweight nested loop generation library, which makes it easy to generate complex, optimized nested loops (such as loops in conv2d).
Halide
Tensor Comprehension