pytorch/xla

Pull Requests Commits

Another update to libtpu, jax, and jaxlib pins to see if they are now working.

yaoshiang committed 93 days ago

1963a338

Update pins again

yaoshiang committed 93 days ago

a718edce

yaoshiang committed 93 days ago

556ecbef

[torchax]: JittableModule statedict handling (#9195)

zmelumian972 committed 94 days ago

Verified afe425e2

Update README.md (#9397)

shauheen committed 96 days ago

Verified 08187535

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

zzzwen committed 96 days ago

Verified d780e55c

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

qihqi committed 96 days ago

Verified 1784a081

[build_developer] Fix vision installation command (#9395)

tengyifei committed 97 days ago

Verified dabcb2dc

Update gru.py to use is_fn_pure (#9393)

tengyifei committed 97 days ago

Verified 86471c07

Update documentations for scan cache (#9394)

iwknow committed 97 days ago

Verified f21f9c0f

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

hfan committed 98 days ago

Verified b4108cb5

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

zmelumian972 committed 98 days ago

Verified 2533c429

Add cache support for scan_layers (#9297)

iwknow committed 98 days ago

Verified 52994af5

Wrap def_static to enable warning reporting. (#9380)

zhanyong-wan committed 98 days ago

Verified 39251247

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

jeffhataws committed 98 days ago

Verified efdf117b

Implement prng_key as a mutable array (#9305)

tengyifei committed 98 days ago

Verified 0ee876f3

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

zhanyong-wan committed 99 days ago

Verified aaff9591

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

qihqi committed 99 days ago

Verified 89cc066a

Update the tuned block size. (#9376)

QiliangCui committed 99 days ago

Verified f20b89f9

Deprecate ShapeOfXlaOp in favor of GetShape (#9381)

zhanyong-wan committed 99 days ago

Verified b737b4a1

Fix race condition in runAtenTest (#9306)

benawilson committed 99 days ago

Verified f8bd6c48

Improve the API for binding python functions in C++ (#9351)

zhanyong-wan committed 99 days ago

Verified 86c59725

update l_ref in kernel matrix calculation (#9372)

Chenyaaang committed 100 days ago

Verified 77c40f3f

Avoid recompilation caused by `scan_layers` (#9367)

tengyifei committed 100 days ago

Verified 41b9e452

fix all_gather_into_tensor test and logic (#9332)

bfolie committed 100 days ago

Verified 9b0b02f0

Improve `assume_pure` SPMD functionality (#9360)

tengyifei committed 100 days ago

Verified 0be0fbdf

Add an option to not use dlpack. (#9304)

qihqi committed 101 days ago

Verified 55a75404

docs: fix contribute guide command (#9355)

giacomoni committed 102 days ago

Verified 20899c72

Update ragged paged attention kernel to prevent vmem oom (#9346)

Chenyaaang committed 103 days ago

Verified bea86eec

Split _ops and _decomps. (#9323)

qihqi committed 103 days ago

Verified 3a1ed628

Older