Remove commented code

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Signed-off-by: Kshitij Lakhani <33047503+KshitijLakhani@users.noreply.github.com>

Author: KshitijLakhani

transformer_engine/jax/cpp_extensions/attention.py

@@ -1542,8 +1542,8 @@ def q_seqlens_for_striped_for_rank(self, q_segment_ids, q_segment_pos, max_segme
         segment_changes = jnp.concatenate(
             [
                 first_is_segment,  # First valid element starts a segment
-                (valid_segment_ids[..., 1:] != valid_segment_ids[..., :-1]) |
-                # ((valid_segment_pos[..., 1:] != valid_segment_pos[..., :-1] + 1) & actual_valid[..., 1:])
+                (valid_segment_ids[..., 1:] != valid_segment_ids[..., :-1])
+                | (valid_segment_pos[..., 1:] != valid_segment_pos[..., :-1] + 1),
                 (valid_segment_pos[..., 1:] != valid_segment_pos[..., :-1] + 1),
             ],
             axis=-1,