lecture-jax

lecture-jax copied to clipboard

MAINT: update njit to jit

5

This PR updates `njit` to `jit` now that `numba>=0.59` is installed https://github.com/QuantEcon/meta/issues/112

mmcky

[multiple_lectures] use correct timing

11

This PR fix #163.

shlff

MAINT: use Github based GPU instance

14

This PR makes use of the `Tesla T4` instance now available on `GitHub Actions` as a beta instance

mmcky

Fixes JAX intro lecture

8

kp992

Use logspace instead of linspace in Aiyagari lecture

1

:wave:

cc7768

Add a new lecture on job search

1

The code to be converted to JAX is here: https://github.com/QuantEcon/book-dp1/blob/main/source_code_py/markov_js.py https://github.com/QuantEcon/book-dp1/blob/main/source_code_py/markov_js_with_sep.py These can be combined into one lecture, with the second set of code as an exercise. Background is in...

jstac

Not timing properly

4

In this lecture, when we do the timings, we do not use `block_until_ready` and we don't run twice to eliminate compile time. https://jax.quantecon.org/opt_savings_1.html I think this is probably true in...

jstac

Right now, in the final figure, OPI is beating HPI for some values of m in the final figure of opt_savings_2 and opt_invest. Can HPI beat OPI if we loosen...

jstac

In place operations in JAX

2

@mmcky Can you please search for this line: In line with immutability, JAX does not support inplace operations: It looks like JAX is sorting in place now. Could you please...

jstac

Add a new Jax lecture ``ifp_advanced_jax`` based on https://python.quantecon.org/ifp_advanced.html The code is ready, and the details need to be filled in (TBD in a new PR).

shlff