-
-
Notifications
You must be signed in to change notification settings - Fork 55
[ge_arrow] Update to JAX #717
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
Conversation
|
📖 Netlify Preview Ready! Preview URL: https://pr-717--sunny-cactus-210e3e.netlify.app (b58cafe) 📚 Changed Lecture Pages: aiyagari, cake_eating_numerical, career, coleman_policy_iter, egm_policy_iter, ge_arrow, harrison_kreps, ifp, ifp_advanced, jv, lake_model, lqcontrol, mccall_correlated, mccall_fitted_vfi, mccall_model, mccall_model_with_separation, mccall_q, odu, optgrowth_fast, two_auctions, wald_friedman_2 |
|
📖 Netlify Preview Ready! Preview URL: https://pr-717--sunny-cactus-210e3e.netlify.app (bf55c41) 📚 Changed Lecture Pages: aiyagari, cake_eating_numerical, career, coleman_policy_iter, egm_policy_iter, ge_arrow, harrison_kreps, ifp, ifp_advanced, jv, lake_model, lqcontrol, mccall_correlated, mccall_fitted_vfi, mccall_model, mccall_model_with_separation, mccall_q, odu, optgrowth_fast, two_auctions, wald_friedman_2 |
|
📖 Netlify Preview Ready! Preview URL: https://pr-717--sunny-cactus-210e3e.netlify.app (253dea6) 📚 Changed Lecture Pages: aiyagari, cake_eating_numerical, career, coleman_policy_iter, egm_policy_iter, ge_arrow, harrison_kreps, ifp, ifp_advanced, jv, lake_model, lqcontrol, mccall_correlated, mccall_fitted_vfi, mccall_model, mccall_model_with_separation, mccall_q, odu, optgrowth_fast, two_auctions, wald_friedman_2 |
|
📖 Netlify Preview Ready! Preview URL: https://pr-717--sunny-cactus-210e3e.netlify.app (be07276) 📚 Changed Lecture Pages: aiyagari, cake_eating_numerical, career, coleman_policy_iter, egm_policy_iter, ge_arrow, harrison_kreps, ifp, ifp_advanced, jv, lake_model, lqcontrol, mccall_correlated, mccall_fitted_vfi, mccall_model, mccall_model_with_separation, mccall_q, odu, optgrowth_fast, two_auctions, wald_friedman_2 |
Updated the
ge_arrow.mdto JAX and complemented the styling consistent with the operation manual.Key changes:
RecurCompetitiveclass as aNamedTuple.compute_rc_modelfunction. Inside this function, arguments of sub-functions can be written in the same way as the definitions in the theory part.jittedthe main computation function, and usedjax.lax.fori_loopto conduct loops.