Worked on enhancing the reliability of distributed communication tests in the facebookresearch/param repository, focusing on the validation of all_gather operations within the Param Communications Trace Replay suite. Addressed a critical bug by correcting data type specifications and refining expected message sizes, ensuring that test cases accurately mirrored the behavior of distributed communication primitives. This targeted debugging effort, implemented in Python and leveraging strong testing skills, improved the stability and accuracy of the continuous integration pipeline. The work contributed to greater confidence in distributed training components by aligning test expectations with actual communication patterns, supporting robust development and maintenance of the codebase.