Skip to content

[WIP] feat: adopt a more JAX-like API - #211

Draft
beckermr wants to merge 18 commits into
mainfrom
jax-api-changes
Draft

beckermr wants to merge 18 commits into
mainfrom
jax-api-changes

Commits

  1. Commits on Mar 23, 2026

  2. Commits on Mar 24, 2026

  3. Commits on Mar 26, 2026

  4. Commits on Apr 7, 2026