Federal budget to pump $2B into AI compute, set up new safety institute