diff --git a/docs/.vitepress/config/shared.ts b/docs/.vitepress/config/shared.ts
index 1f4961d2..26ba1292 100644
--- a/docs/.vitepress/config/shared.ts
+++ b/docs/.vitepress/config/shared.ts
@@ -24,6 +24,9 @@ export const shared = defineConfig({
 
   markdown: {
     math: true,
+    // Load these languages ahead of time for embedded code highlighting
+    // TODO: remove this when Shiki fixes markdown embedded language loading
+    languages: ['js', 'ts', 'html'],
     codeTransformers: [
       // We use `[!!code` in demo to prevent transformation, here we revert it back.
       {