Migrate AMSS lecture from NumPy/Numba to JAX with automatic differentiation#244

Closed

Copilot wants to merge 10 commits intomainfrom

copilot/fix-243

Commits on Aug 15, 2025

Initial plan
Copilot
committed
Create JAX-based utility functions and interpolation utilities for AMSS model

Copilot
and
HumphreyYang
committed
Complete JAX migration of AMSS lecture with working examples

Copilot
and
HumphreyYang
committed
Fix function naming and test complete JAX AMSS workflow

Copilot
and
HumphreyYang
committed
add jax install
HumphreyYang
committed
Fix JAX dependency in environment.yml and missing AMSSParams import

Copilot
and
mmcky
committed
Fix import issue in jax_amss_simple.py for Jupyter Book execution context

Copilot
and
mmcky
committed
Fix NameError: government_spending not defined in AMSS lecture

Copilot
and
mmcky
committed
Fix jax_amss_simple.py execution issue in Jupyter Book context

Copilot
and
mmcky
committed
Fix undefined variables in AMSS lecture - solution and simulation models

Copilot
and
mmcky
committed