18. Execution Statistics#

This table contains the latest execution statistics.

Document

Modified

Method

Run Time (s)

Status

aiyagari_jax

2023-12-21 00:44

cache

56.55

arellano

2023-12-21 00:45

cache

34.88

cake_eating_numerical

2023-12-21 00:45

cache

22.54

ifp_egm

2023-12-21 00:47

cache

117.05

intro

2023-12-21 00:47

cache

1.06

inventory_dynamics

2023-12-21 00:54

cache

437.16

jax_intro

2023-12-21 00:55

cache

34.03

kesten_processes

2023-12-21 00:55

cache

25.02

markov_asset

2023-12-21 00:56

cache

25.02

mle

2023-12-21 00:56

cache

17.78

newtons_method

2023-12-21 00:59

cache

168.08

opt_invest

2023-12-21 01:00

cache

40.68

opt_savings

2023-12-21 01:00

cache

26.02

short_path

2023-12-21 01:00

cache

8.96

status

2023-12-21 01:00

cache

3.8

troubleshooting

2023-12-21 00:47

cache

1.06

wealth_dynamics

2023-12-21 01:01

cache

30.41

zreferences

2023-12-21 00:47

cache

1.06

These lectures are built on linux instances through github actions and amazon web services (aws) to enable access to a gpu. These lectures are built on a p3.2xlarge that has access to 8 vcpu's, a V100 NVIDIA Tesla GPU, and 61 Gb of memory.

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
Thu Dec 21 01:00:47 2023       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.182.03   Driver Version: 470.182.03   CUDA Version: 12.1     |
|-------------------------------+----------------------+----------------------+
| 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 V100-SXM2...  Off  | 00000000:00:1E.0 Off |                    0 |
| N/A   33C    P0    39W / 300W |    310MiB / 16160MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
+-----------------------------------------------------------------------------+