Refactor the Pallas launcher to make it easier to operate on Jax Arrays as well as TorchTPU tensors - #3190
Merged
Merged
Meta CodeSync / Meta Import Checks
succeeded
Jul 31, 2026 in 0s
All checks passed, this Pull Request can be imported!
This check validates that the Pull Request can be imported to Meta internal infrastructure
Loading