Take a look at Tiramisu Polyhedral Compiler

[lukstafi]

https://arxiv.org/abs/1804.10694

Tiramisu uses a four-level intermediate representation that allows full separation between the algorithms, loop transformations, data layouts, and communication.