Skip to content

fix(deps): split jax[cuda] into cpu and cuda extras to support CPU-only installs (#443) - #539

Open
udbhav-shrinet wants to merge 1 commit into
google-research:masterfrom
udbhav-shrinet:fix/cpu-only-installs
Open

udbhav-shrinet wants to merge 1 commit into
google-research:masterfrom
udbhav-shrinet:fix/cpu-only-installs

Commits

  1. Commits on Oct 13, 2026