depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile. - View it on GitHub