The best way to ensure public gains from the automation wave would be a VAT so that people and companies just pay the tax when they buy things or employ services. For businesses, it gets baked into the cost of production at every level. It makes it much harder for large companies, which are experts at reducing their taxes, to benefit from the American infrastructure and citizenry without paying into it.