Tax companies at 2x the single-taxpayer withholding amount lost per employee displaced.
So if an employee normally makes 100,000 per year and the average withholding is ~35,000, if 10 jobs are displaced, the tax charge is 700,000. Company still saves 300k, and the AI displacing those ten heads probably Costa 10k per license anyway, so there's still room for AI to make economic sense while backstopping the sociological consequences.
Put that tax revenue directly towards social services or minimum monthly reimbursement for everyone (UBI, just with a different name)