pytorch/xla

Pull Requests Commits

implement diagonal_copy (#9416)

Chenyaaang committed 168 days ago

Verified ffce7fb8

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

qihqi committed 168 days ago

Verified 1da73090

Add support for 3.13 builds (#9417)

bhavya01 committed 168 days ago

Verified 468760a0

Style improvements. (#9410)

zhanyong-wan committed 169 days ago

Verified 37c8eb16

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

bhavya01 committed 169 days ago

Verified cafff837

Update CODEOWNERS (#9409)

bhavya01 committed 170 days ago

Verified 96bfad65

adding tolerance for numeric test of checkpointing (#9404)

yaoshiang committed 170 days ago

Verified 193f7049

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

jeffhataws committed 171 days ago

Verified 7e3efc58

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

laithsakka committed 171 days ago

Verified 926700d7

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

ysiraichi committed 171 days ago

Verified 10ed5547

Migrate to correct logger interface (#9191)

emmanuel-ferdman committed 171 days ago

Verified a36d3e54

Support torch.distributed.scatter collective (#9365)

bfolie committed 171 days ago

Verified 01db65db

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

qihqi committed 172 days ago

Verified 2d81349c

[torchax]: JittableModule statedict handling (#9195)

zmelumian972 committed 173 days ago

Verified afe425e2

Update README.md (#9397)

shauheen committed 175 days ago

Verified 08187535

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

zzzwen committed 175 days ago

Verified d780e55c

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

qihqi committed 175 days ago

Verified 1784a081

[build_developer] Fix vision installation command (#9395)

tengyifei committed 176 days ago

Verified dabcb2dc

Update gru.py to use is_fn_pure (#9393)

tengyifei committed 176 days ago

Verified 86471c07

Update documentations for scan cache (#9394)

iwknow committed 177 days ago

Verified f21f9c0f

Make the shape property adhere to torch.Tensor interface. (#9342)

hfan committed 177 days ago

Verified b4108cb5

Torchax: JittableModule isinstance to work with encapsulated model (#9375)

zmelumian972 committed 177 days ago

Verified 2533c429

Add cache support for scan_layers (#9297)

iwknow committed 177 days ago

Verified 52994af5

Wrap def_static to enable warning reporting. (#9380)

zhanyong-wan committed 177 days ago

Verified 39251247

Revert "Avoid unnecessary copy in TensorSource (#8849)" (#9379)

jeffhataws committed 177 days ago

Verified efdf117b

Implement prng_key as a mutable array (#9305)

tengyifei committed 178 days ago

Verified 0ee876f3

Include both pytorch and torch_xla revisions in the compilation cache key. (#9383)

zhanyong-wan committed 178 days ago

Verified aaff9591

Add autocast feature as `torchax.amp.autocast`. (#9364)

qihqi committed 178 days ago

Verified 89cc066a

Update the tuned block size. (#9376)

QiliangCui committed 178 days ago

Verified f20b89f9

Deprecate ShapeOfXlaOp in favor of GetShape (#9381)

zhanyong-wan committed 178 days ago

Verified b737b4a1

Newer Older