Really cool. That's how this project got started. I was doing the exact thing you were doing (tile-based RL path finding). But it was soooo slow, so I started implementing it in tensors, and that's when I realized I could do a whole game of Snake.