add starter code

Author: komaksym

questions/195_gradient-checkpointing/starter_code.py

@@ -1,4 +1,15 @@
+import numpy as np
+
 # Implement your function below.
+def checkpoint_forward(funcs, input_arr):
+    """
+    Applies a list of functions in sequence to the input array, simulating gradient checkpointing by not storing intermediates.
+
+    Args:
+        funcs (list of callables): List of functions to apply in sequence.
+        input_arr (np.ndarray): Input numpy array.
 
-def your_function(...):
+    Returns:
+        np.ndarray: The output after applying all functions, same shape as output of last function.
+    """
     pass