pytorch/xla

Pull Requests Commits

Fix metrics saver when a file location is specified (#8644)

yaochengji committed 279 days ago

Verified 7243b808

Build v2.6 release wheel (#8645)

tengyifei committed 279 days ago

Verified 01839974

Add a lowering to linear instead of using torch decomp (#8643)

qihqi committed 279 days ago

Verified 25d09fdd

Prepare v2.6.0-rc10 (#8639)

tengyifei committed 280 days ago

Verified 310d2d81

scan documentation (#8631)

tengyifei committed 280 days ago

Verified 225c65bd

Fix grad accumulation tests on TPUv5/v6 (#8628)

tengyifei committed 281 days ago

Verified 8e6ca600

Rename torch_xla2 to torchax (#8599)

qihqi committed 283 days ago

Verified 8b241409

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

vealocia committed 283 days ago

Verified 79e4e72f

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

bhavya01 committed 283 days ago

Verified aef5f6ea

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

jeffhataws committed 283 days ago

Verified e694c90d

Prepare to cut r2.6 rc9 (#8619)

tengyifei committed 284 days ago

Verified 6f367dfa

Improve build_developer.sh (#8579)

tengyifei committed 285 days ago

Verified d382feed

use env to skip PJRT initialize (#8609)

zpcore committed 285 days ago

Verified 557d9f34

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

fenghuizhang committed 285 days ago

Verified 9a1405d6

Introducing experimental gradient accumulation API (#8584)

rpsilva-aws committed 285 days ago

Verified 36dcba32

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

fenghuizhang committed 286 days ago

Verified 5b877beb

Add two placeholder example directories (#8601)

tengyifei committed 286 days ago

Verified fbbdfca1

training UX: automatic generating make_train_step (#8495)

qihqi committed 286 days ago

Verified f5b33c5d

Prepare to create 2.6.0rc8 builds (#8598)

tengyifei committed 286 days ago

Verified bbd66b93

Pipes attn_logits_soft_cap through multi_queries_paged_attention (#8583)

fenghuizhang committed 286 days ago

Verified 6b785fc8

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

ManfeiBai committed 290 days ago

Verified a295f7d0

Re-introduce "XLA_USE_32BIT_LONG" flag (#8571)

rpsilva-aws committed 292 days ago

Verified 78fa7576

Prepare to make rc6 wheel (#8582)

tengyifei committed 292 days ago

Verified 18efb6dc

[scan] Reduce memory usage (#8562)

tengyifei committed 292 days ago

Verified dc768796

Update terraform providers for tpu-pytorch (#8575)

tengyifei committed 293 days ago

Verified 02602094

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

rpsilva-aws committed 293 days ago

Verified 1d615566

Fix mark_sharding logic (#8578)

yaochengji committed 293 days ago

Verified 14acab7a

Correct libtpu_storage_path for stable libtpu builds (#8576)

tengyifei committed 293 days ago

Verified 79de480f

Prepare to make rc5 wheel (#8573)

tengyifei committed 293 days ago

Verified e513e132

Update iam.auto.tfvars (#8570)

tengyifei committed 293 days ago

Verified 4edbf61d

Newer Older