pytorch/xla

Pull Requests Commits

Remove obsolete TODOs

bhavya01 committed 116 days ago

baafc892

cli.print_chip_info() causes segfault. Comment for now

bhavya01 committed 116 days ago

98752f54

Fix CPU tests for python 3.12 (#9443)

bhavya01 committed 116 days ago

Verified 5d05f0a3

Change update_deps script so that latest stable version can be pulled instead of latest nightly (#9424)

bfolie committed 118 days ago

Verified 66ad14f9

Update TPU CI container image (#9434)

bhavya01 committed 123 days ago

Verified 1c00dea2

Add JAX dependency for Python 3.12 (#9438)

tengyifei committed 123 days ago

Verified 53ea5862

Error Handling: refactor `XlaCoordinator` to use status types. (#9386)

ysiraichi committed 124 days ago

Verified 5b8ddb12

Add Build Trigger for 2.8-rc1 release (#9425)

pgmoka committed 124 days ago

Verified 752ddba7

Update CI docker images to use 3.12 (#9427)

bhavya01 committed 124 days ago

Verified 773ecdfb

Optimize w8a8 quantized matmul kernel (#9412)

vanbasten23 committed 124 days ago

Verified 4101ea56

Install libtpu directly instead of torch_xla[tpu] (#9423)

bhavya01 committed 125 days ago

Verified 9c8ae9f9

`EmbeddingDenseBackward`: Remove `padding_idx` cast to `double` (#9406)

unterumarmung committed 125 days ago

Verified 3d92aa75

s/torch_xla2/torchax (#9353)

tengyifei committed 127 days ago

Verified 9499e6f4

implement diagonal_copy (#9416)

Chenyaaang committed 127 days ago

Verified ffce7fb8

Pin update to 20250617. After that there is a pallas regression (#9415)

qihqi committed 127 days ago

Verified 1da73090

Add support for 3.13 builds (#9417)

bhavya01 committed 128 days ago

Verified 468760a0

Style improvements. (#9410)

zhanyong-wan committed 129 days ago

Verified 37c8eb16

Add nightly and dev images for python 3.12 (#9408)

bhavya01 committed 129 days ago

Verified cafff837

Update CODEOWNERS (#9409)

bhavya01 committed 130 days ago

Verified 96bfad65

adding tolerance for numeric test of checkpointing (#9404)

yaoshiang committed 130 days ago

Verified 193f7049

Allgather coalescee: Check tuple shape only if return shape is tuple. (#9403)

jeffhataws committed 130 days ago

Verified 7e3efc58

Prepare for pytorch tensor impl change in is_contiguous_custom (#9402)

laithsakka committed 130 days ago

Verified 926700d7

ErrorHandling: make `GetComputationClient()` return `StatusOr<T>` type. (#9384)

ysiraichi committed 131 days ago

Verified 10ed5547

Migrate to correct logger interface (#9191)

emmanuel-ferdman committed 131 days ago

Verified a36d3e54

Support torch.distributed.scatter collective (#9365)

bfolie committed 131 days ago

Verified 01db65db

Revert "[torchax]: JittableModule statedict handling" (#9401)

qihqi committed 132 days ago

Verified 2d81349c

[torchax]: JittableModule statedict handling (#9195)

zmelumian972 committed 133 days ago

Verified afe425e2

Update README.md (#9397)

shauheen committed 135 days ago

Verified 08187535

Add jax_device context manager to control the device target (#9382)

zzzwen committed 135 days ago

Verified d780e55c

Refactor call_jax to allow implementing ops in python (#9354)

qihqi committed 135 days ago

Verified 1784a081

Older