lecture-jax icon indicating copy to clipboard operation
lecture-jax copied to clipboard

[NHM] Implement Kesten Processses with Smaller Memory Footprint on GPU

Open HumphreyYang opened this issue 2 years ago • 1 comments
trafficstars

As discussed in #37.

From @jstac: I feel there might be a useful middle ground here. For example, use lax.scan to run forward 100 time periods and then call that 5 or 10 times in an outer loop in pure Python.

HumphreyYang avatar May 08 '23 04:05 HumphreyYang

Thanks @HumphreyYang . Applicable to Kesten process lecture, inventory lecture, wealth dynamics lecture.

jstac avatar May 08 '23 04:05 jstac