All pull requests

Pull requests

• fix: add missing Float8 tensor type mappings to StableHLO to prevent process-wide abort

  #9764·hassanaymanelprens58-maker opened Apr 30, 2026·

  ·

• security: fix integer overflow in as_strided and OOM in JSON parsing

  #9763·hassanaymanelprens58-maker opened Apr 29, 2026·

  ·

• Fix composite pass missing captured values in ops with nested regions

  #9760·mstojkovicTT opened Mar 30, 2026·

  ·
  2

• Fix persistent compilation cache not working on TPU

  #9759·shr1ram opened Mar 23, 2026·

  ·
  1

• Fix scan_layers to support layers with no weights

  #9758·kredd2506 opened Mar 21, 2026·

  ·
  1

• Make a function wrapper for _get_xla_sharding_specs

  #9757·ajakovljevicTT opened Mar 17, 2026·

  ·
  3

• Optimize as_strided_copy fast path to support offset

  #9728·Zantares opened Dec 10, 2025·

  ·

• Expose torch_xla.runtime.clear_computation_cache binding to clear computation cache

  #9720·jameszianxuTT opened Nov 24, 2025·

  ·
  5

• Fix typo in warning message for SPMD mode

  #9718·ddilbazTT opened Nov 20, 2025·

  ·

• implement XLAHooks and register it to PyTorch when loaded.

  #9683·iwknow opened Oct 20, 2025·

  ·
  1

  4

• Ensuring PJRT Client destroy/destructor is called

  #9675·saarthak-aws opened Oct 7, 2025·

  ·
  13

• Add destructor for XLAGraphExecutor to free resource

  #9636·Zantares opened Sep 11, 2025·

  ·
  2

• Test ci

  #9633·bhavya01 opened Sep 10, 2025·

  ·

• Adding functionality for metadata communication across hosts

  #9570·kvshbg-aws opened Aug 19, 2025·

  ·
  8

• xla: relax error message assertion in dot_general test (allow backend…

  #9557·Rahulchaube1 opened Aug 13, 2025·

  ·
  2

• XLAShardedTensor.to_local() support

  #9505·Hoomaaan opened Jul 24, 2025·

  ·
  10

• Add tests for DTensor.from_local() support for XLA device mesh

  #9504·Hoomaaan opened Jul 24, 2025·

  ·

• Support SPMD placeholder tensors

  #9489·rpsilva-aws opened Jul 17, 2025·

  ·

• Update torchprime tests to Python 3.12

  bugtesting
  #9481·pgmoka opened Jul 15, 2025·

  ·
  

• feat: abstraction of xla::OpSharding proto using wrapper class

  #9467·kvshbg-aws opened Jul 10, 2025·

  ·
  25

• Introduce multi-operand collective permute

  #9450·rpsilva-aws opened Jul 8, 2025·

  ·

• Convert some XLA_CHECKs to fatal errors.

  #9439·zhanyong-wan opened Jul 2, 2025·

  ·

• Add a test on OOM error

  #9414·zhanyong-wan opened Jun 26, 2025·

  ·

• [wip]ruff inter on torchax

  #9396·zzzwen opened Jun 20, 2025·

  ·
  1

• [WIP] Add autocast in torchax

  #9361·qihqi opened Jun 16, 2025·

  ·