issues
search
pytorch
/
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.46k
stars
467
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Remove linalg.matrix_rank from skiplist (#7497)
#8096
simonteozw
opened
4 hours ago
0
Fix lgamma, mvlgamma, multinomial, and nanmedian (#7513)
#8095
simonteozw
opened
6 hours ago
0
Enable cross entropy loss for xla autocast with FP32 precision (#7992)
#8094
avizon-aws
opened
1 day ago
0
[benchmarks] Default to functionalization disabled.
#8093
ysiraichi
opened
1 day ago
0
Fix torch.min and torch.mode (#7513)
#8092
simonteozw
opened
1 day ago
0
Implement `torch.ops.aten.embedding_renorm_`
#8091
guyao
opened
2 days ago
0
[WIP] Implement upsample_bilinear2d_aa
#8090
barney-s
opened
2 days ago
0
[2.5 release] GPU docker image failed to run mnist test
#8089
ManfeiBai
opened
2 days ago
1
Is this content still relevant?
#8088
mikegre-google
opened
2 days ago
1
Add new_empty
#8087
tracych477
closed
2 days ago
0
[hard] `nn.functional.max_pool2d` and `nn.functional.max_pool3d`
#8086
ManfeiBai
opened
2 days ago
0
Reduced the max_unpoolxd logic into one function #7524
#8085
hosseinsarshar
closed
2 days ago
0
Added the support for max_unpool1d, max_unpool2d, and max_unpool3d #7524
#8084
hosseinsarshar
closed
2 days ago
0
Update trigger to 2.5.0-rc5 for 2.5 release
#8083
ManfeiBai
closed
2 days ago
0
R2.5
#8082
ManfeiBai
closed
2 days ago
0
Add support for erfinv
#8081
matinehAkhlaghinia
closed
2 days ago
0
Update README.md to 2.6.0.dev for nightly wheel
#8080
ManfeiBai
closed
2 days ago
0
Pin update
#8079
qihqi
opened
3 days ago
0
Force github workflow catch PyTorch commit from release/2.5 branch for PyTorch/XLA `r2.5` branch
#8078
ManfeiBai
closed
3 days ago
2
cherry pick collective op all_to_all_single
#8077
zpcore
closed
3 days ago
1
Removes linalg.eigvalsh from skiplist and fixes errors that occur using custom atol/rtol.
#8076
anishfish2
closed
4 days ago
0
Removed linalg.cond from skiplist
#8075
anishfish2
closed
4 days ago
0
P2P send recv test gives errors
#8074
ajayvohra2005
opened
4 days ago
3
add `argwhere`
#8073
ManfeiBai
opened
4 days ago
0
Fixed linalg.eigh by adding lowering and adding custom atol/rtol
#8072
anishfish2
closed
4 days ago
0
Optimizer Memory in AdamW/Adam vs SGD
#8071
dangthatsright
opened
4 days ago
1
Run the verifier on Torchbench in the Nigthly CI.
#8070
ysiraichi
opened
4 days ago
2
Check warning msg in second captured warning.
#8069
zhuhaozhe
opened
5 days ago
0
Avoids error removing linalg.eig from skiplist by modifying atol and rtol
#8068
anishfish2
closed
5 days ago
0
Use Tuple instead of tuple in scan.py
#8067
tengyifei
closed
5 days ago
0
Use Tuple instead of tuple in scan.py
#8066
tengyifei
closed
5 days ago
1
Upstream PyTorch 2.5 release branch's XLA CI is failing
#8065
JackCaoG
closed
4 days ago
5
Support dist.all_to_all_single
#8064
zpcore
closed
4 days ago
1
Support the xla_force_host_platform_device_count for CPU for multiple device count and GSPMD
#8063
mario-aws
opened
5 days ago
1
change r2.5 to use jax stable version 0.4.33
#8062
ManfeiBai
opened
5 days ago
0
R2.5 jaxstable
#8061
ManfeiBai
closed
5 days ago
0
R2.5 jax stable version
#8060
ManfeiBai
closed
5 days ago
0
Poor performance with 1 GPU?
#8059
Patataman
opened
5 days ago
2
[torch_xla2] Fix op_info test for scatter_reduce
#8058
dvhg
closed
5 days ago
0
PjRtComputationClient::ExecuteReplicated core dump when encountering a scalar
#8057
mars1248
opened
5 days ago
3
Support FSDPv2 compute dtype
#8056
lausannel
opened
5 days ago
4
Update trigger to 2.5.0-rc3 for 2.5 release
#8055
ManfeiBai
closed
5 days ago
0
Print the training update after the device execution
#8054
JackCaoG
opened
6 days ago
0
Update tests for Neuron
#8053
jeffhataws
closed
5 days ago
0
[cherry-pick] Update Openxla-pin to Sep13, libtpu-pin to Sep13, jax to 0.4.33_nightly
#8052
ManfeiBai
closed
2 days ago
0
R2.5
#8051
ManfeiBai
closed
6 days ago
0
[cherry-pick] Follow 2 releases grace period for API deprecation (#8003)
#8050
ManfeiBai
closed
6 days ago
0
How to run XLA with CPU offloaded models
#8049
radna0
opened
6 days ago
10
[torch_xla2] Fix nn.functional.conv2d and conv3d
#8048
dvhg
closed
6 days ago
0
[feat] Request for conda build
#8047
Mon-ius
opened
1 week ago
0
Next