Skip to content

v0.9.0 2025/11/06: Multi-device SPMD execution; TPU installation support; Deprecating `xlabuilder`.

Choose a tag to compare

View all tags
@janpfeifer janpfeifer released this 06 Nov 07:51
· 41 commits to main since this release
v0.9.0
1947897
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

v0.9.0 2025/11/06: Multi-device SPMD execution; TPU installation support; Deprecating xlabuilder.

  • gopjrt_installer: Added TPU installation.
  • Package pjrt:
    • Renamed Client.Devices to Client.AllDevices.
    • Added Client.NumDevices to get the number of addressable devices.
    • Added Compile().WithSPMD for multi-device SPMD execution.
    • Default compilation is "portable" (can be run in any device), but it allows for device specific as well.
    • Arena pools for powers-of-2 sizes.
  • Replaced xlabuilder with stablehlo in examples:
    • Enabled darwin tests using stablehlo
  • Updated PJRT C API and protos from XLA sources.
Assets 5
Loading