Treesitter in-math detect?

[medwatt]

I have been growing increasingly frustrated with the default latex syntax highlight in neovim. I would rather turn it off if it wasn't for the fact that I use the function below to trigger autosnippets. The default syntax highlighting makes scrolling extremely sluggish. Treesitter highlighting has no such issue. I was wondering if anyone knows of way to detect math zones with treesitter.

    function M.in_math()
        return vim.api.nvim_eval("vimtex#syntax#in_mathzone()") == 1
    end

[evesdropper]

pretty sure there are a few implementations that exist online already - a few I recall looking at recently at are in a snippets plugin by iurimateus and dotfiles by kunzaatko