GitHub

Original file line numberDiff line numberDiff line change

@@ -29,8 +29,7 @@ kernelspec:

2929

```

3030
3131

```{seealso}

32-

**GPU:** A version of this lecture which makes use of [jax](https://jax.readthedocs.io) to run the code

33-

on a `GPU` is [available here](https://jax.quantecon.org/newtons_method.html)

32+

A version of this lecture using [JAX](https://github.com/jax-ml/jax) is {doc}`available here <jax:newtons_method>`

3433

```

3534
3635

## Overview

@@ -788,7 +787,7 @@ With the larger overhead, the speed is not better than the optimized `scipy` fun

788787

Our next step is to investigate a large market with 3,000 goods.

789788
790789

A JAX version of this section using GPU accelerated linear algebra and

791-

automatic differentiation is available [here](https://jax.quantecon.org/newtons_method.html#application)

790+

automatic differentiation is {doc}`available here <jax:newtons_method>`

792791
793792

The excess demand function is essentially the same, but now the matrix $A$ is $3000 \times 3000$ and the parameter vectors $b$ and $c$ are $3000 \times 1$.

794793

Read the original on github.com ↗