18. Execution Statistics#

This table contains the latest execution statistics.

Document

Modified

Method

Run Time (s)

Status

aiyagari_jax

2023-09-22 00:25

cache

58.99

arellano

2023-09-22 00:26

cache

34.68

cake_eating_numerical

2023-09-22 00:26

cache

22.54

ifp_egm

2023-09-22 00:27

cache

72.86

intro

2023-09-22 00:28

cache

1.43

inventory_dynamics

2023-09-22 00:34

cache

412.11

jax_intro

2023-09-22 00:35

cache

36.47

kesten_processes

2023-09-22 00:36

cache

35.71

markov_asset

2023-09-22 00:36

cache

23.84

mle

2023-09-22 00:36

cache

19.58

newtons_method

2023-09-22 00:39

cache

164.08

opt_invest

2023-09-22 00:40

cache

39.98

opt_savings

2023-09-22 00:40

cache

24.8

short_path

2023-09-22 00:40

cache

11.02

status

2023-09-22 00:40

cache

4.09

troubleshooting

2023-09-22 00:28

cache

1.43

wealth_dynamics

2023-09-22 00:41

cache

35.03

zreferences

2023-09-22 00:28

cache

1.43

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
Fri Sep 22 00:40:52 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   30C    P0    38W / 300W |    310MiB / 16160MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
+-----------------------------------------------------------------------------+