Code

Implemented jax.lax.while primitive#16

Code

Merged

patrick-kidger merged 5 commits intopatrick-kidger:mainpatrick-kidger/quax:mainfrom ymahlau:mainymahlau/quax:mainCopy head branch name to clipboardJul 21, 2024

+226

Commits

Commits on Apr 20, 2024

• Implemented jax.lax.while primitive

  ymahlau

  committed
  c37cbdd

  Copy full SHA for c37cbdd

Commits on May 4, 2024

• extended unitful example

  ymahlau

  committed
  ac6300f

  Copy full SHA for ac6300f

• added test cases for while implementation

  ymahlau

  committed
  cf9d3bf

  Copy full SHA for cf9d3bf

Commits on Jul 14, 2024

• test grad over closure

  ymahlau

  committed
  504471d

  Copy full SHA for 504471d

Commits on Jul 20, 2024

• minor changes to while primitive

  ymahlau

  committed
  d58a300

  Copy full SHA for d58a300