India has earmarked significant investments in artificial intelligence, global capability centres and digital public infrastructure while boosting training and education in its latest budget.