pytorch/xla

Pull Requests Commits

Update README.md

pgmoka committed 271 days ago

Verified 9baaa543

Update documentation.md

pgmoka committed 278 days ago

Verified 307eb1ff

Update README.md

pgmoka committed 278 days ago

Verified 9bd75685

Add information about codegen in codegen/xla_native_functions.yaml (#8817)

pgmoka committed 278 days ago

Verified 76b0ce50

Add info related to codegen (#8834)

pgmoka committed 278 days ago

Verified 5dd50099

Expose mark_sharding_with_gradients as a public API. (#8826)

iwknow committed 279 days ago

Verified 5caaaee8

Fix for triton pin update. (#8825)

ysiraichi committed 280 days ago

Verified 818c92df

Add build triggers for 2.7-rc1 (#8828)

zpcore committed 280 days ago

Verified fed01200

Revamp SPMD guide (#8807)

tengyifei committed 281 days ago

Verified 5beb6511

Adapt Neuron data type tests (#8835)

rpsilva-aws committed 282 days ago

Verified bd46c0ea

Remove xla_mark_sharding_dynamo_custom_op (#8823)

tengyifei committed 282 days ago

Verified 27d455aa

Support unused arguments in xb.call_jax (#8830)

tengyifei committed 283 days ago

Verified b3f38757

[torchax] Fix functional.max_pool (#8814)

dvhg committed 283 days ago

Verified 0aac10e2

Proper test coverage for Neuron CollectivePermute (#8818)

rpsilva-aws committed 283 days ago

Verified acc4c739

Fix jax_env_context (#8819)

bhavya01 committed 283 days ago

Verified 2e04cfb4

Update MarkShardingFunction to be AOTAutograd traceable (#8821)

tengyifei committed 283 days ago

Verified c969276e

[Compilation Cache] Add API to query number of cached computation graphs (#8822)

lsy323 committed 283 days ago

Verified bdf30f66

Refactor Computation of jax into creation and call (#8789)

qihqi committed 284 days ago

Verified 53b3ab81

Remove pages_per_seq padding constrains and add MHA support (#8816)

bythew3i committed 284 days ago

Verified af16c210

Fix the ragged kernel bug (#8806)

vanbasten23 committed 285 days ago

Verified 6804c65f

Document xla override (#8802)

pgmoka committed 286 days ago

Verified 0407d6d1

Leverage placeholder tensors for GA API (#8808)

rpsilva-aws committed 286 days ago

Verified 5b7462c6

Manually register einsum on xla (#8801)

pgmoka committed 289 days ago

Verified 1db6c861

Misc fixes (#8804)

qihqi committed 289 days ago

Verified 7aea0585

Re-enable triton tests. (#8803)

ysiraichi committed 289 days ago

Verified 2c70a1c2

Lower `_conj_copy` and `alias` operation. (#8686)

ysiraichi committed 290 days ago

Verified 6c53a1ef

Fix ragged paged attention v2 to resolve the recompilation issue (#8797)

yaochengji committed 290 days ago

Verified d06a9c97

Pin update to 20250303 (#8788)

tengyifei committed 290 days ago

Verified 9b61c1ad

Use placeholder tensor in scan (#8785)

tengyifei committed 290 days ago

Verified a9279287

Use .backward() with in-place grad mutations for the GA API (#8768)

rpsilva-aws committed 290 days ago

Verified 4540d817

Older