Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions lectures/_admonition/gpu.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```
9 changes: 1 addition & 8 deletions lectures/aiyagari_jax.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,14 +13,7 @@ kernelspec:

# The Aiyagari Model

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

## Overview
Expand Down
9 changes: 1 addition & 8 deletions lectures/arellano.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,14 +13,7 @@ kernelspec:

# Default Risk and Income Fluctuations

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

In addition to what's in Anaconda, this lecture will need the following libraries:
Expand Down
9 changes: 1 addition & 8 deletions lectures/inventory_dynamics.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,14 +23,7 @@ kernelspec:

# Inventory Dynamics

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

```{index} single: Markov process, inventory
Expand Down
9 changes: 1 addition & 8 deletions lectures/jax_intro.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,7 @@ kernelspec:
# An Introduction to JAX


```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

This lecture provides a short introduction to [Google JAX](https://github.com/google/jax).
Expand Down
9 changes: 1 addition & 8 deletions lectures/kesten_processes.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,14 +29,7 @@ kernelspec:
:depth: 2
```

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

In addition to what's in Anaconda, this lecture will need the following libraries:
Expand Down
9 changes: 1 addition & 8 deletions lectures/newtons_method.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,7 @@ kernelspec:

# Newton’s Method via JAX

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

## Overview
Expand Down
9 changes: 1 addition & 8 deletions lectures/opt_invest.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,7 @@ kernelspec:

# Optimal Investment

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

We require the following library to be installed.
Expand Down
9 changes: 1 addition & 8 deletions lectures/opt_savings.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,14 +13,7 @@ kernelspec:

# Optimal Savings

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

In addition to what’s in Anaconda, this lecture will need the following libraries:
Expand Down
9 changes: 1 addition & 8 deletions lectures/short_path.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,14 +15,7 @@ kernelspec:

# Shortest Paths

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

## Overview
Expand Down
9 changes: 1 addition & 8 deletions lectures/wealth_dynamics.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,7 @@ kernelspec:

# Wealth Distribution Dynamics

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and JAX for GPU programming.

Free GPUs are available on Google Colab. To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support. If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

This lecture is the extended JAX implementation of [this lecture](https://python.quantecon.org/wealth_dynamics.html).
Expand Down