Grey's parents are Amity; no one knows that - he doesn't tell anyone obviously.

So he knows his way around there in the dark. So, yes, when the train reaches the end of the line, he jumps first, into the dark, waiting for Tris.