23. Execution Statistics#

This table contains the latest execution statistics.

Document

Modified

Method

Run Time (s)

Status

aiyagari_jax

2024-06-13 03:47

cache

107.59

arellano

2024-06-13 03:47

cache

20.33

hopenhayn

2024-06-13 03:49

cache

120.93

ifp_egm

2024-06-13 03:51

cache

118.92

intro

2024-06-13 03:51

cache

0.95

inventory_dynamics

2024-06-13 03:52

cache

36.66

inventory_ssd

2024-06-13 04:24

cache

1911.61

jax_intro

2024-06-13 04:24

cache

35.49

job_search

2024-06-13 04:25

cache

8.24

kesten_processes

2024-06-13 04:25

cache

12.9

lucas_model

2024-06-13 04:25

cache

15.11

markov_asset

2024-06-13 04:25

cache

11.45

mle

2024-06-13 04:25

cache

14.01

newtons_method

2024-06-13 04:28

cache

124.47

opt_invest

2024-06-13 04:28

cache

22.96

opt_savings_1

2024-06-13 04:29

cache

32.63

opt_savings_2

2024-06-13 04:29

cache

21.03

overborrowing

2024-06-13 04:31

cache

106.22

short_path

2024-06-13 04:31

cache

3.04

status

2024-06-13 04:31

cache

2.02

troubleshooting

2024-06-13 03:51

cache

0.95

wealth_dynamics

2024-06-13 04:34

cache

218.77

zreferences

2024-06-13 03:51

cache

0.95

These lectures are built on linux instances through github actions that has access to a gpu. These lectures make use of the nvidia T4 card.

You can check the backend used by JAX using:

import jax
# Check if JAX is using GPU
print(f"JAX backend: {jax.devices()[0].platform}")
JAX backend: gpu

and the hardware we are running on:

!nvidia-smi
/opt/conda/envs/quantecon/lib/python3.11/pty.py:89: RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock.
  pid, fd = os.forkpty()
Thu Jun 13 04:31:12 2024       
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.54.03              Driver Version: 535.54.03    CUDA Version: 12.5     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  Tesla T4                       On  | 00000001:00:00.0 Off |                  Off |
| N/A   61C    P0              37W /  70W |    107MiB / 16384MiB |      1%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+
                                                                                         
+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
+---------------------------------------------------------------------------------------+