For syntax highlighting of .mll and .mly files (plus lots of other features) use the ocaml-platform extension.
Fork of https://github.com/hackwaly/vscode-ocaml to isolate highlighting support for just OCamllex (.mll) and OCamlyacc/Menhir (.mly) files