Results 4 issues of Justin Fu

The fallback trajectory causes bugs with sim agents ignoring collisions and driving with constant velocities as seen in https://github.com/waymo-research/waymax/issues/52. This is because when the sim agent outputs invalid actions, the...

### Description The `unsafe_rbg` fold_in function is not sensitive to the order in which data is folded in. The underlying cause is that `unsafe_rbg` derives its key based on `key...

bug