Quantitative Economics with JAX

This website presents a set of lectures on quantitative economic modeling using GPUs and Google JAX. The lectures are designed and written by Thomas J. Sargent and John Stachurski.

Introduction

• 1. About
• 2. An Introduction to JAX
• 3. Newton’s Method via JAX
• 4. An Asset Pricing Problem

Simulation

• 5. Inventory Dynamics
• 6. Kesten Processes and Firm Dynamics
• 7. Wealth Distribution Dynamics

Data and Empirics

• 8. Maximum Likelihood Estimation

Dynamic Programming

• 9. Shortest Paths
• 10. Optimal Investment
• 11. Optimal Savings
• 12. Endogenous Grid Method
• 13. Default Risk and Income Fluctuations
• 14. The Aiyagari Model
• 15. Cake Eating: Numerical Methods

Other

• 16. Troubleshooting
• 17. References
• 18. Execution Statistics