QuantEcon / QuantEcon/lecture-python-advanced.myst
[TEST] Migrate AMSS to JAX
Open
Nobody has claimed this yet.
- Dominant language
- TeX
- Stars
- 37
- Forks
- 26
- Avg merge
- 13d 22h
- Merged PRs (30d)
- 2
Description
The aim of this issue is to migrate the amss lecture to use JAX.
Specifically:
- Follow the guidelines listed at https://manual.quantecon.org/styleguide/jax-conversion.html.
- 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 properNamedTuplestructure, following the guidelines. - Avoid using methods within classes. Move them out as pure functions that take parameter classes as arguments.
- Correct any typos and mathematical errors in the lecture.
- 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.
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.