Fix import issue in jax_amss_simple.py for Jupyter Book execution context

Co-authored-by: mmcky <8263752+mmcky@users.noreply.github.com>

Author: Copilot

lectures/_static/lecture_specific/amss/jax_amss_simple.py

@@ -6,12 +6,9 @@
 import jax.numpy as jnp
 from jax import jit, grad, vmap
 from typing import NamedTuple
-try:
-    from .jax_utilities import *
-    from .jax_interpolation import *
-except ImportError:
-    from jax_utilities import *
-    from jax_interpolation import *
+
+# Note: jax_utilities and jax_interpolation are loaded via :load: directives 
+# before this file, so their functions are available in the global namespace
 
 
 class AMSSSimpleParams(NamedTuple):