When the federal government began extending trillions of dollars in aid to companies and investors battered by the pandemic, some of the country’s biggest private equity firms have lobbied hard to shape the stimulus in their favor.