The National Science Foundation, Department of Energy and Department of Commerce would get some of the highest investments for artificial intelligence-related work under the latest budget released by the White House.