Releases: QuantEcon/lecture-jax
Releases · QuantEcon/lecture-jax
publish-2026jan14
What's Changed
Full Changelog: publish-2025dec27...publish-2026jan14
publish-2025dec27
What's Changed
- ⬆️ Upgrade build software and dependencies by @mmcky in #274
- ⬆️ Bump actions/upload-artifact from 5 to 6 by @dependabot[bot] in #273
- ⬆️ Bump dawidd6/action-download-artifact from 11 to 12 by @dependabot[bot] in #275
- ⬆️ Bump sphinx-reredirects from 1.0.0 to 1.1.0 by @dependabot[bot] in #276
Full Changelog: publish-2025dec09...publish-2025dec27
publish-2025dec09
What's Changed
- BUG: Remove x_validate and y_validate as they're unnecessary by @blkboxeconomist in #266
- MAINT: Add exercise_style configuration and update quantecon-book-theme by @mmcky in #272
New Contributors
- @blkboxeconomist made their first contribution in #266
Full Changelog: publish-2025dec4.3...publish-2025dec09
publish 2025 dec 4
Remove num_layers from Config and improve visualization (#265) - Remove redundant num_layers field from Config class - Update Keras model builder to use len(config.layer_sizes) - Format MSE values to 6 decimal places in summary table - Change plots to show validation data instead of training data - Use red color with alpha=0.5 for scatter plots 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-authored-by: Claude <[email protected]>
publish 2025 dec 5
publish-2025dec4.2 Refactor train_jax_model to use fori_loop and simplify return value (…
publish 2025 dec 4
JIT compile train_network and optimize performance (#263)
* Add type hints and inline comments to function signatures in ifp_dl.md
Updated function signatures to include type hints and inline comments
for improved code readability and consistency:
- initialize_layer: Added type hints for in_dim, out_dim, and key
- initialize_network: Added type hints for key and layer_sizes
- forward: Added type hints for params and a (asset level)
- compute_lifetime_value: Added type hints for all parameters
- K: Added type hints and removed Args/Returns from docstring
- solve_model: Added type hints and removed Args/Returns from docstring
- compute_lifetime_value_ifp: Added type hints and removed Args/Returns
Also fixed grammar in initialize_layer docstring ("a the" → "the").
🤖 Generated with [Claude Code](https://claude.com/claude-code)
Co-Authored-By: Claude <[email protected]>
* JIT compile train_network and optimize performance in ifp_dl.md
Refactored train_network for JIT compilation with significant performance
improvements:
- Added @jax.jit decorator with static_argnames for config and loss_fn
- Replaced Python for loop with jax.lax.fori_loop for JIT compatibility
- Removed print statements (not JIT-compatible)
- Used value_history.at[i].set() to accumulate history in-place
- Replaced if statements with jnp.where for conditional logic
- Used jax.tree.map for conditionally updating best_params PyTree
Performance optimizations:
- Reduced path_length from 320 to 200 (35% faster)
- Reduced num_paths from 220 to 100 (additional speedup)
- Added warmup calls to trigger JIT compilation before timing
- Added block_until_ready() for accurate timing measurements
Results with new configuration:
- Cake eating model: 0.49s (was ~0.82s)
- IFP model: 3.91s (was ~10s+)
🤖 Generated with [Claude Code](https://claude.com/claude-code)
Co-Authored-By: Claude <[email protected]>
---------
Co-authored-by: Claude <[email protected]>
publish-2025dec04b
What's Changed
Full Changelog: publish-2025dec04...publish-2025dec04b
publish-2025dec04
What's Changed
Full Changelog: publish-2025dec01d...publish-2025dec04
publish-2025dec01d
What's Changed
Full Changelog: publish-2025dec01c...publish-2025dec01d
publish-2025dec01c
Full Changelog: publish-2025dec01b...publish-2025dec01c