Quantitative Economics with JAX

This website presents a set of lectures on quantitative economic modeling using GPUs and Google JAX.

Introduction

• 1. About
• 2. An Introduction to JAX
• 3. Newton’s Method via JAX

Simulation

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

Asset Pricing

• 7. Asset Pricing: The Lucas Asset Pricing Model
• 8. An Asset Pricing Problem

Dynamic Programming

• 9. Job Search
• 10. Optimal Savings I: Value Function Iteration
• 11. Optimal Savings II: Alternative Algorithms
• 12. Shortest Paths
• 13. Optimal Investment
• 14. Inventory Management Model
• 15. Endogenous Grid Method
• 16. Cake Eating: Numerical Methods

Macroeconomic Models

• 17. Default Risk and Income Fluctuations
• 18. The Aiyagari Model
• 19. The Hopenhayn Entry-Exit Model
• 20. Bianchi Overborrowing Model

Data and Empirics

• 21. Maximum Likelihood Estimation

Other

• 22. Troubleshooting
• 23. References
• 24. Execution Statistics