In May 2025, Ankit Gulati enhanced the pytorch/pytorch repository by delivering cross-device compatibility testing for the FSDP2 module. He refactored existing test cases to be device-agnostic, enabling validation across CUDA, HPU, and XPU accelerators. This work involved restructuring Python-based unit tests and integrating distributed computing principles to ensure consistent behavior on multiple hardware platforms. By automating multi-accelerator testing within the continuous integration pipeline, Ankit improved the reliability and speed of distributed training feature validation. His contributions deepened test coverage for FSDP2, reducing device-specific issues and supporting faster feature readiness without introducing new bugs during the development period.