If your loop is gpu bound, it’s going to end up unrolled in a warp anyway... When we get to a point that machine learning is dominated by tree traversal, than python will be toast.