AI infrastructure platform for running AI & ML workloads