Nsys JAX update for remapping nccl nodes

[Steboss]

Updated data_loaders.py so we can:

• find all the NCCL range IDs
• delete the NCCL rows
• remap the IDs correctly to preserve the correct ids for the logic XlaModule --> Thunk: Operation

[olupton]

There seem to be a bunch of new test failures.

[Steboss]

@olupton
it looks like the latest PRs solved the issue here

[olupton]

Thanks! I think it should be OK not to update the RangeStack values at this point.