Worked on the Lightning-AI/torchmetrics repository to address stability issues related to division operations on Apple Silicon hardware. Focused on debugging and code refactoring in Python, the developer delivered a targeted fix that updated the _safe_divide function to disable non-blocking division when the device type is set to 'mps'. This change resolved runtime errors encountered on MPS devices, improving reliability for users running PyTorch workloads on Apple Silicon. The solution was implemented with careful attention to cross-device compatibility and was tracked through a dedicated pull request, ensuring clear traceability and alignment with the repository’s quality and review standards.