Code

Enable JAX backend#812

Code

Open

timmens wants to merge 5 commits intomainiza-institute-of-labor-economics/gettsim:mainfrom jaxiza-institute-of-labor-economics/gettsim:jaxCopy head branch name to clipboard

+4,026-6,023

Commits

Commits on Jan 21, 2025

• Fix test_config.py

  timmens

  committed
  1bfdbc0

  Copy full SHA for 1bfdbc0

• Actually use JAX to vectorize function

  timmens

  committed
  6d19b9d

  Copy full SHA for 6d19b9d

Commits on Jan 22, 2025

• Create and update JAX and CUDA pixi environments

  timmens

  committed
  e5bccc3

  Copy full SHA for e5bccc3

• Add jit flag to compute_taxes_and_transfers and create test case that jits

  timmens

  committed
  27992a2

  Copy full SHA for 27992a2

Commits on Jan 23, 2025

• Actually use make_vectorizable and align testing

  timmens

  committed
  d0bc497

  Copy full SHA for d0bc497