Neural Topological Ordering for Computation Graphs