pytorch/xla

Pull Requests Commits

Make partition_spec as tuple to fix issue #8522 (#8613)

vealocia committed 241 days ago

Verified 79e4e72f

Update README.md to recommend torch_xla cxx11 builds by default. (#8624)

bhavya01 committed 242 days ago

Verified aef5f6ea

Improve the nprocs value error message to be actionable (#8622)

jeffhataws committed 242 days ago

Verified e694c90d

Prepare to cut r2.6 rc9 (#8619)

tengyifei committed 243 days ago

Verified 6f367dfa

Improve build_developer.sh (#8579)

tengyifei committed 243 days ago

Verified d382feed

use env to skip PJRT initialize (#8609)

zpcore committed 243 days ago

Verified 557d9f34

Fix the signature of paged_attention by marking megacore_mode optional (#8615)

fenghuizhang committed 243 days ago

Verified 9a1405d6

Introducing experimental gradient accumulation API (#8584)

rpsilva-aws committed 244 days ago

Verified 36dcba32

Fix an issue when piping attn_logits_soft_cap through in vllm. (#8600)

fenghuizhang committed 244 days ago

Verified 5b877beb

Add two placeholder example directories (#8601)

tengyifei committed 244 days ago

Verified fbbdfca1

training UX: automatic generating make_train_step (#8495)

qihqi committed 244 days ago

Verified f5b33c5d

Prepare to create 2.6.0rc8 builds (#8598)

tengyifei committed 244 days ago

Verified bbd66b93

Pipes attn_logits_soft_cap through multi_queries_paged_attention (#8583)

fenghuizhang committed 245 days ago

Verified 6b785fc8

prepare rc7 for r2.6 with libtpu update (#8586)

ManfeiBai committed 249 days ago

Verified a295f7d0

Re-introduce "XLA_USE_32BIT_LONG" flag (#8571)

rpsilva-aws committed 250 days ago

Verified 78fa7576

Prepare to make rc6 wheel (#8582)

tengyifei committed 250 days ago

Verified 18efb6dc

[scan] Reduce memory usage (#8562)

tengyifei committed 251 days ago

Verified dc768796

Update terraform providers for tpu-pytorch (#8575)

tengyifei committed 251 days ago

Verified 02602094

Improve and refine MLP tests for extensibility and A/B testing (#8561)

rpsilva-aws committed 251 days ago

Verified 1d615566

Fix mark_sharding logic (#8578)

yaochengji committed 251 days ago

Verified 14acab7a

Correct libtpu_storage_path for stable libtpu builds (#8576)

tengyifei committed 251 days ago

Verified 79de480f

Prepare to make rc5 wheel (#8573)

tengyifei committed 252 days ago

Verified e513e132

Update iam.auto.tfvars (#8570)

tengyifei committed 252 days ago

Verified 4edbf61d

pin update (#8559)

lsy323 committed 253 days ago

Verified 1c89675e

Set gradient_as_bucket_view=True in test_train_mp_imagenet (#8558)

tengyifei committed 253 days ago

Verified 2fdf721f

fix batch_norm amp autocast (#8556)

yaochengji committed 253 days ago

Verified fda4ecae

[scan] Test we don't recompile under debugging env flags (#8555)

tengyifei committed 253 days ago

Verified 79a7de0c

Introduce HLO graph bindings (#8551)

rpsilva-aws committed 255 days ago

Verified 3784b268

Always run TPU tests on new PRs (#8552)

tengyifei committed 256 days ago

Verified 6963e195

Metadata agnostic user computation hash (#8550)

rpsilva-aws committed 256 days ago

Verified 6e5659ec

Newer Older