Skip to content

Releases: QuantEcon/lecture-jax

publish-2026jan14

14 Jan 03:37
0f8f62d

Choose a tag to compare

What's Changed

  • Update GPU admonition to clarify lectures run on CPUs by @mmcky in #278

Full Changelog: publish-2025dec27...publish-2026jan14

publish-2025dec27

27 Dec 05:52
8c06965

Choose a tag to compare

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

09 Dec 01:38
c390704

Choose a tag to compare

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

Full Changelog: publish-2025dec4.3...publish-2025dec09

publish 2025 dec 4

04 Dec 12:49
102113e

Choose a tag to compare

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

04 Dec 11:25
9724c6c

Choose a tag to compare

publish-2025dec4.2

Refactor train_jax_model to use fori_loop and simplify return value (…

publish 2025 dec 4

04 Dec 08:57
b99eb52

Choose a tag to compare

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

04 Dec 08:11
d311b01

Choose a tag to compare

What's Changed

  • Optimize JAX JIT compilation strategy in neural network lecture by @jstac in #262

Full Changelog: publish-2025dec04...publish-2025dec04b

publish-2025dec04

04 Dec 04:10
e544012

Choose a tag to compare

What's Changed

  • Update jax_nn.md with improved data handling and exercise strategies by @jstac in #261

Full Changelog: publish-2025dec01d...publish-2025dec04

publish-2025dec01d

01 Dec 05:38
e4b026d

Choose a tag to compare

What's Changed

  • Update quantecon-book-theme to 0.14.0 by @mmcky in #260

Full Changelog: publish-2025dec01c...publish-2025dec01d

publish-2025dec01c

01 Dec 02:23

Choose a tag to compare