Skip to content

Refactor the Pallas launcher to make it easier to operate on Jax Arrays as well as TorchTPU tensors - #3190

Merged
AmesingFlank merged 1 commit into
mainfrom
AmesingFlank/stack/104
Jul 31, 2026
Merged

Refactor the Pallas launcher to make it easier to operate on Jax Arrays as well as TorchTPU tensors#3190
AmesingFlank merged 1 commit into
mainfrom
AmesingFlank/stack/104