Direct die cooling