Hacktoberfest 2026: the issues maintainers tagged for October, open and beginner-friendly. Browse Hacktoberfest issues

[TEST] Migrate AMSS to JAX

Open
  • Is something wrong?
#243 2 comments 1 reaction 1 assignee View on GitHub

Nobody has claimed this yet.

  • #244 by @copilot-swe-agent — closed without merging

Assessment

This issue has not been assessed yet.

Description

The aim of this issue is to migrate the amss lecture to use JAX.

Specifically:

  1. Follow the guidelines listed at https://manual.quantecon.org/styleguide/jax-conversion.html.
  2. Break up large classes and use autodifferentiation for marginal utility calculations. Some of the classes are not currently included in the lecture (e.g., AMSS classes and utility classes). These can be found in lectures/_static/lecture_specific/amss. Please add them to the lecture using a proper NamedTuple structure, following the guidelines.
  3. Avoid using methods within classes. Move them out as pure functions that take parameter classes as arguments.
  4. Correct any typos and mathematical errors in the lecture.
  5. Follow the existing lecture setup, keep the implementation concise, and ensure the code closely matches the mathematical notation. Optimize the code using vectorization and JIT whenever possible.
Dominant language
TeX
Stars
37
Forks
27
Avg merge
2d 4h
Merged PRs (30d)
1

Getting set up

This project ships no dev container, Dockerfile or contributing guide, so setting up is up to you: start from its README, and see our first-contribution guide for the general steps.

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from QuantEcon/lecture-python-advanced.myst

All issues in QuantEcon/lecture-python-advanced.myst

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.