Looks pretty similar to me[1]

[1]: https://nvidia.github.io/tilus/getting-started/tutorials/mat...