lecture-jax

lecture-jax copied to clipboard

Next lecture for JAX conversion

6

**EDIT** transferred from intermediate lectures. The discussion is still relevant. Perhaps this lecture is a good candidate for attempting to port to JAX: https://python.quantecon.org/ifp.html I think it will be challenging...

jstac

There are a number of places where citations and references have used `md` links We need to harmonise and use the proper `cite` role as per: https://manual.quantecon.org/styleguide/references.html

mmcky

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

1

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...

HumphreyYang

Timed code should be run twice in `inventory_dynamics`. Cut code cell after the timing and then start a new cell for the plots.

jstac

[cake_eating_numerical] Numba time less than JAX

4

The lecture `cake_eating_numerical` was removed in #175 as the `jax` time was longer than `numba`. We need to review the implementations in this lecture as a sense check. A copy...

mmcky

TST: Full build

2

mmcky

TST: runs-on service for GPU

2

mmcky

[jax_intro] Update `%time` magic with `%timeit`

7

This PR fixes #206.

HumphreyYang

Enable software version checking

3

This PR just adds software versions to the `status` page

mmcky

Replace `%time` with `%timeit`

3

@shizejin reported that there is a misleading comparison in the introduction to JAX:  It might relates to the timing magic we used. `%time` is sometimes influenced other factors such...

HumphreyYang