I think this is the same program; it gives an example

https://www.deepmind.com/blog/competitive-programming-with-a...