Skip to content

Commit 1c36519

Browse files
jstacclaude
andcommitted
Update mccall_model.md: Convert from Numba to JAX implementation
Converted the McCall job search model lecture from using Numba to JAX for better performance and modern functional programming approach. Key changes: - Replaced Numba's @jit and @jitclass with JAX's @jax.jit - Converted NumPy arrays to JAX arrays (jnp) - Used NamedTuple instead of jitclass for model classes - Implemented JAX-style while_loop for iterations - Added vmap for efficient vectorized computations - Updated all code examples and exercises to use JAX The conversion maintains all functionality while providing improved performance and compatibility with modern Python scientific computing stack. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <[email protected]>
1 parent 9992ef5 commit 1c36519

File tree

1 file changed

+234
-251
lines changed

1 file changed

+234
-251
lines changed

0 commit comments

Comments
 (0)