AI for AMR prediction