Bug Description

This is related to an issue found on mujoco_warp:
google-deepmind/mujoco_warp#458

I could track it to the bitonic sort.
The issue is however non-deterministic and I could not get a minimal repro.

System Information

No response