Hi,

I came here through the annotated_diffusion notebook, which I really like.
Through the notebook I realized that Google Colab and VSCode render inline math differently.

From my understanding, the preferred way in markdown is to write inline math with single $ and math blocks with double $$.
However, in the annotated_diffusion notebook, inline math is escaped with \\( and \\), which does not render correctly in VSCode.

Would it be possible to change the inline math syntax to single $, as preferred for markdown?