Skip to content

Commit 7dfe7ee

Browse files
committed
Convert wealth_dynamics lecture from Numba to JAX
- Add JAX imports and GPU acceleration warning - Convert @jitclass WealthDynamics to NamedTuple + factory pattern - Replace @jit functions with pure JAX functions using jax.lax.scan - Implement explicit JAX random key management - Update all application code and exercises to use JAX implementation - Follow QuantEcon JAX style guide rules for functional programming
1 parent df9484a commit 7dfe7ee

File tree

1 file changed

+220
-174
lines changed

1 file changed

+220
-174
lines changed

0 commit comments

Comments
 (0)