diff --git a/compiler/packages/babel-plugin-react-forget/src/Entrypoint/Program.ts b/compiler/packages/babel-plugin-react-forget/src/Entrypoint/Program.ts
index 27ec7a1a4d..593347b0c0 100644
--- a/compiler/packages/babel-plugin-react-forget/src/Entrypoint/Program.ts
+++ b/compiler/packages/babel-plugin-react-forget/src/Entrypoint/Program.ts
@@ -265,6 +265,19 @@ export function compileProgram(
         }
       },
 
+      FunctionExpression(
+        fn: NodePath<t.FunctionExpression>,
+        pass: CompilerPass
+      ): void {
+        if (!shouldVisitNode(fn, pass)) {
+          return;
+        }
+
+        if (compileAndInsertNewFunctionDeclaration(fn, pass) === true) {
+          hasForgetMutatedOriginalSource = true;
+        }
+      },
+
       ArrowFunctionExpression(
         fn: NodePath<t.ArrowFunctionExpression>,
         pass: CompilerPass
diff --git a/compiler/packages/babel-plugin-react-forget/src/HIR/BuildHIR.ts b/compiler/packages/babel-plugin-react-forget/src/HIR/BuildHIR.ts
index 7510bb4edc..2edcf18e32 100644
--- a/compiler/packages/babel-plugin-react-forget/src/HIR/BuildHIR.ts
+++ b/compiler/packages/babel-plugin-react-forget/src/HIR/BuildHIR.ts
@@ -83,6 +83,8 @@ export function lower(
   let id: string | null = null;
   if (func.isFunctionDeclaration() && func.get("id").node != null) {
     id = (func.get("id") as NodePath<t.Identifier>).node.name;
+  } else if (func.isFunctionExpression() && func.get("id").node != null) {
+    id = (func.get("id") as NodePath<t.Identifier>).node.name;
   }
   const params: Array<Place> = [];
   func.get("params").forEach((param) => {
diff --git a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/infer-function-expression-component.expect.md b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/infer-function-expression-component.expect.md
new file mode 100644
index 0000000000..4bc8b55c44
--- /dev/null
+++ b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/infer-function-expression-component.expect.md
@@ -0,0 +1,31 @@
+
+## Input
+
+```javascript
+// @compilationMode(infer)
+
+const Component = function ComponentName(props) {
+  return <Foo />;
+};
+
+```
+
+## Code
+
+```javascript
+import { unstable_useMemoCache as useMemoCache } from "react"; // @compilationMode(infer)
+
+const Component = function ComponentName(props) {
+  const $ = useMemoCache(1);
+  let t0;
+  if ($[0] === Symbol.for("react.memo_cache_sentinel")) {
+    t0 = <Foo />;
+    $[0] = t0;
+  } else {
+    t0 = $[0];
+  }
+  return t0;
+};
+
+```
+      
\ No newline at end of file
diff --git a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/infer-function-expression-component.js b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/infer-function-expression-component.js
new file mode 100644
index 0000000000..7fe4c9c0cc
--- /dev/null
+++ b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/infer-function-expression-component.js
@@ -0,0 +1,5 @@
+// @compilationMode(infer)
+
+const Component = function ComponentName(props) {
+  return <Foo />;
+};
diff --git a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-9a47e97b5d13.expect.md b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-9a47e97b5d13.expect.md
index 978a78e0c1..d14c4d9de7 100644
--- a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-9a47e97b5d13.expect.md
+++ b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-9a47e97b5d13.expect.md
@@ -14,11 +14,23 @@ const FancyButton = React.forwardRef(function (props, ref) {
 ## Code
 
 ```javascript
-// Valid because hooks can be used in anonymous function arguments to
+import { unstable_useMemoCache as useMemoCache } from "react"; // Valid because hooks can be used in anonymous function arguments to
 // forwardRef.
 const FancyButton = React.forwardRef(function (props, ref) {
+  const $ = useMemoCache(3);
   useHook();
-  return <button {...props} ref={ref} />;
+  const c_0 = $[0] !== props;
+  const c_1 = $[1] !== ref;
+  let t0;
+  if (c_0 || c_1) {
+    t0 = <button {...props} ref={ref} />;
+    $[0] = props;
+    $[1] = ref;
+    $[2] = t0;
+  } else {
+    t0 = $[2];
+  }
+  return t0;
 });
 
 ```
diff --git a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-e66a744cffbe.expect.md b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-e66a744cffbe.expect.md
index df628834cc..25ed1c43d4 100644
--- a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-e66a744cffbe.expect.md
+++ b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-e66a744cffbe.expect.md
@@ -14,11 +14,23 @@ const FancyButton = forwardRef(function (props, ref) {
 ## Code
 
 ```javascript
-// Valid because hooks can be used in anonymous function arguments to
+import { unstable_useMemoCache as useMemoCache } from "react"; // Valid because hooks can be used in anonymous function arguments to
 // forwardRef.
 const FancyButton = forwardRef(function (props, ref) {
+  const $ = useMemoCache(3);
   useHook();
-  return <button {...props} ref={ref} />;
+  const c_0 = $[0] !== props;
+  const c_1 = $[1] !== ref;
+  let t0;
+  if (c_0 || c_1) {
+    t0 = <button {...props} ref={ref} />;
+    $[0] = props;
+    $[1] = ref;
+    $[2] = t0;
+  } else {
+    t0 = $[2];
+  }
+  return t0;
 });
 
 ```
diff --git a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-eacfcaa6ef89.expect.md b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-eacfcaa6ef89.expect.md
index 6682b758c9..e37104ccc6 100644
--- a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-eacfcaa6ef89.expect.md
+++ b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-eacfcaa6ef89.expect.md
@@ -14,11 +14,21 @@ const MemoizedFunction = memo(function (props) {
 ## Code
 
 ```javascript
-// Valid because hooks can be used in anonymous function arguments to
+import { unstable_useMemoCache as useMemoCache } from "react"; // Valid because hooks can be used in anonymous function arguments to
 // memo.
 const MemoizedFunction = memo(function (props) {
+  const $ = useMemoCache(2);
   useHook();
-  return <button {...props} />;
+  const c_0 = $[0] !== props;
+  let t0;
+  if (c_0) {
+    t0 = <button {...props} />;
+    $[0] = props;
+    $[1] = t0;
+  } else {
+    t0 = $[1];
+  }
+  return t0;
 });
 
 ```
diff --git a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-fe6042f7628b.expect.md b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-fe6042f7628b.expect.md
index 60cbea92b8..1c676d9825 100644
--- a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-fe6042f7628b.expect.md
+++ b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-fe6042f7628b.expect.md
@@ -2,6 +2,7 @@
 ## Input
 
 ```javascript
+// @compilationMode(infer)
 // This is valid because "use"-prefixed functions called in
 // unnamed function arguments are not assumed to be hooks.
 unknownFunction(function (foo, bar) {
@@ -15,6 +16,7 @@ unknownFunction(function (foo, bar) {
 ## Code
 
 ```javascript
+// @compilationMode(infer)
 // This is valid because "use"-prefixed functions called in
 // unnamed function arguments are not assumed to be hooks.
 unknownFunction(function (foo, bar) {
diff --git a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-fe6042f7628b.js b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-fe6042f7628b.js
index 189de4201d..a480e74a83 100644
--- a/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-fe6042f7628b.js
+++ b/compiler/packages/babel-plugin-react-forget/src/__tests__/fixtures/compiler/rules-of-hooks/rules-of-hooks-fe6042f7628b.js
@@ -1,3 +1,4 @@
+// @compilationMode(infer)
 // This is valid because "use"-prefixed functions called in
 // unnamed function arguments are not assumed to be hooks.
 unknownFunction(function (foo, bar) {