Skip to content

publish-2025 dec 2

Choose a tag to compare

@jstac jstac released this 02 Dec 11:28
60235eb
Fix bugs in ifp_advanced lecture (#757)

- Fix typo: jnp.block_until_ready(a_start) → a_star.block_until_ready()
- Add missing functools.partial import for JAX JIT static arguments
- Fix compute_asset_stationary JIT compilation with static_argnums
- Remove invalid verbose parameter from solve_model calls
- Refactor solve_model to use jax.lax.while_loop for better performance
- Clean up code structure and improve function signatures

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-authored-by: Claude <noreply@anthropic.com>