Skip to content

JAX: Add JAX implementation for wealth dynamics #304

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
jstac merged 11 commits into from
Jan 17, 2023
Merged

JAX: Add JAX implementation for wealth dynamics #304

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
e6699ee
add jax implementation for wealth dynamics time series
Smit-create Dec 30, 2022
9e1657e
Add update cross section in jax
Smit-create Dec 30, 2022
a23e692
fix an error
Smit-create Dec 31, 2022
2cb8379
add a cell to check backend
Smit-create Dec 31, 2022
ff9d4c6
Apply suggestions
Smit-create Jan 4, 2023
063367f
Add for-loop implementation
Smit-create Jan 4, 2023
d942383
Hide numba code
Smit-create Jan 4, 2023
e43c237
Hide all numba code and fix typos
Smit-create Jan 10, 2023
975be52
Remove hidden tags and use glue
Smit-create Jan 13, 2023
6b49755
Add GPU warning
Smit-create Jan 16, 2023
d4e56ba
Update GPU warning
Smit-create Jan 17, 2023
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
7 changes: 5 additions & 2 deletions lectures/status.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,5 +16,8 @@ This table contains the latest execution statistics.
```{nb-exec-table}
```

These lectures are built on `linux` instances through `github actions` so are
executed using the following [hardware specifications](https://docs.github.com/en/actions/using-github-hosted-runners/about-github-hosted-runners#supported-runners-and-hardware-resources)
(status:machine-details)=

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](https://aws.amazon.com/ec2/instance-types/p3/)
that has access to `8 vcpu's`, a `V100 NVIDIA Tesla GPU`, and `61 Gb` of memory.
Loading