Minor issue, but I just noticed using [numerical definitions from nx](https://github.com/elixir-nx/nx/tree/main/nx#numerical-definitions) breaks syntax highlighting for function definitions. # Actual No syntax highlighting for `defn`-defined function: <img width="298" alt="Screen Shot 2021-07-09 at 8 14 33 AM" src="https://user-images.githubusercontent.com/8541241/125083316-d6948400-e08d-11eb-9fb3-6ca302bf89ee.png"> # Expected Syntax highlighting matches that of normal function: <img width="294" alt="Screen Shot 2021-07-09 at 8 14 49 AM" src="https://user-images.githubusercontent.com/8541241/125083593-368b2a80-e08e-11eb-9ecf-49372b7bcb26.png"> These screenshots are from a kitty terminal using neovim 0.5.0 stable and the [gruvbox colorscheme](https://github.com/gruvbox-community/gruvbox).