You can install this extension from the Marketplace.
What about .mdx and .md files?
By default the TDX language is applied only to .tdx files. If TDX files in your project end with .mdx or .md, you can tell VS Code that by adding the following to your workspace settings: