Skip to content
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

Bump the Version of Jax and Jaxlib to 0.4.34 #98

Merged
merged 2 commits into from
Nov 6, 2024
Merged

Bump the Version of Jax and Jaxlib to 0.4.34 #98

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
2fa214d
Bump Jax and Jaxlib Version to 0.4.34
mj023 Oct 23, 2024
8aad630
Add jax/jaxlib version recommendation to README; update jax version i…
timmens Nov 6, 2024
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
2 changes: 1 addition & 1 deletion .pre-commit-config.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ repos:
- id: mypy
files: src|tests
additional_dependencies:
- jax>=0.4.20
- jax>=0.4.34
- numpy
- packaging
- pandas-stubs
Expand Down
14 changes: 10 additions & 4 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,17 +5,23 @@
[![image](https://codecov.io/gh/OpenSourceEconomics/lcm/branch/main/graph/badge.svg)](https://codecov.io/gh/OpenSourceEconomics/lcm)

This package aims to generalize and facilitate the specification, solution, and
estimation of dynamic choice models.
simulation of discrete-continuous dynamic choice models.

## Installation

`LCM` currently needs to be installed from GitHub and requires `jax` and `jaxlib`. If
you aim to run `LCM` on a GPU, you need to install `jaxlib` with CUDA support (for
`LCM` currently needs to be installed from GitHub and requires `jax` and `jaxlib`.

> [!IMPORTANT]
> We recommend installing at least version "0.4.34" of `jax` and `jaxlib`, due to memory
> issues with previous versions.

If you aim to run `LCM` on a GPU, you need to install `jaxlib` with CUDA support (for
Linux) or with support for AMD GPUs / ARM-based Silicon GPUs (for MacOS). In any case,
for installation of `jax` and `jaxlib`, please consult the `jax`
[docs](https://jax.readthedocs.io/en/latest/installation.html#supported-platforms).

> **Note:** We currently test GPU support for `LCM` only on Linux with CUDA 12.
> [!NOTE]
> We currently test GPU support for `LCM` only on Linux with CUDA 12.

## License

Expand Down
Loading
Loading