issues
search
pytorch
/
xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.49k
stars
480
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Implement `torchvision.ops.roi_align` in torchxla2
#8288
qihqi
opened
1 month ago
0
enable tensorsolve
#8287
barney-s
closed
3 weeks ago
0
disable GPU tests and build
#8286
JackCaoG
closed
1 month ago
0
R2.5
#8285
ManfeiBai
closed
1 month ago
0
[DO NOT REVIEW] Experiment the pl.debug_print in Pallas.
#8284
vanbasten23
opened
1 month ago
0
[cherry-pick] Update README.md (#8281)
#8283
ManfeiBai
closed
1 month ago
0
remove debug prints in lu_unpack implementation
#8282
barney-s
closed
1 month ago
0
Update README.md fo 2.5
#8281
ManfeiBai
closed
1 month ago
1
Add torch_pin to 2.5 release branch
#8280
JackCaoG
closed
1 month ago
0
Add torch_pin to 2.5 release branch
#8279
JackCaoG
closed
1 month ago
0
Implement `torchvision.ops.nms` in torch_xla2.
#8278
qihqi
closed
3 weeks ago
0
Update collective op doc based on feedback
#8277
zpcore
closed
1 month ago
0
Update README.md for PyTorch/XLA 2.5 release
#8276
ManfeiBai
closed
1 month ago
0
Fix support for `torch.var_mean`.
#8275
mrguenther
closed
1 month ago
2
[cherrypick] add dist op doc (#8273)
#8274
zpcore
closed
1 month ago
2
add dist op doc
#8273
zpcore
closed
1 month ago
0
Update artifacts.auto.tfvars for 2.5.0 final wheel
#8272
ManfeiBai
closed
1 month ago
0
Cannot move tensors to cpu when in a xmp spawn process
#8271
radna0
opened
1 month ago
4
Clarify that torch_xla2 is only recommended for inference
#8270
cloudchrischan
opened
1 month ago
1
introduce torch_xla2.compile API, make sdxl to use it
#8269
qihqi
closed
1 month ago
0
make 2.5 whl pypi compatible
#8268
lsy323
closed
1 month ago
0
Update XLA pin to 10/16
#8267
JackCaoG
closed
3 weeks ago
6
Add support for `torch.unravel_index`.
#8266
mrguenther
closed
1 month ago
0
Add peak_bytes_used to the get_memory_info
#8265
JackCaoG
closed
1 month ago
1
Update artifacts.auto.tfvars for 2.5.0-rc10
#8264
ManfeiBai
closed
1 month ago
0
Fix op info test for linalg.lu_factor and linalg.ldl_factor
#8263
dvhg
closed
1 month ago
0
Implement lu_unpack in jax
#8262
barney-s
closed
1 month ago
0
Add support for linalg.lstsq
#8261
matinehAkhlaghinia
closed
1 month ago
0
Fix torch digamma (#7383)
#8260
simonteozw
closed
1 month ago
0
Set weights_only to False for xser.load
#8259
JackCaoG
closed
1 month ago
0
Add `unique`, `unique_consecutive`
#8258
mrguenther
closed
1 month ago
4
Avoid running tpu_info cli test on Trillium
#8257
bhavya01
closed
1 month ago
0
Fix nn.functional.bilinear (#7517)
#8256
simonteozw
closed
1 month ago
0
Implement avg_pool1d, avg_pool2d, and avg_pool3d (#7517)
#8255
simonteozw
closed
1 month ago
0
enable lu, lu_solve testing
#8254
barney-s
closed
1 month ago
0
Fix op info test for linalg.inv and linalg.inv_ex
#8253
dvhg
closed
1 month ago
0
Revert "Fix op info test for linalg.inv and linalg.inv_ex (#8239)"
#8252
qihqi
closed
1 month ago
0
Op info test for `linalg.solve_ex .. linalg.tensorinv` (#7504)
#8251
yenkwang
closed
1 month ago
0
enable linalg.vector_norm tests
#8249
barney-s
closed
1 month ago
0
Fixed tests for basic_gnn_edgecnn and basic_gnn_gcn
#8248
anishfish2
closed
1 month ago
1
Publish Python 3.11 Torch XLA wheels
#8250
metrizable
opened
1 month ago
2
Enabling log_normal tests
#8247
barney-s
closed
1 month ago
0
reorganized files and toc
#8246
mikegre-google
closed
1 month ago
4
Improve documentation for `get_memory_info`
#8245
miladm
opened
1 month ago
0
add profiler hints in paralloader
#8244
JackCaoG
closed
1 month ago
0
update runtime error message for minibatch
#8243
JackCaoG
closed
2 weeks ago
0
Fix `test_arange_nan` after introduction of checks to `out=` functions.
#8242
ysiraichi
closed
2 weeks ago
2
XLA2 does not support maxpool
#8241
Chaosruler972
opened
1 month ago
2
XLA2 does not work with jax 0.4.34 (but did work on jax 0.4.33)
#8240
Chaosruler972
opened
1 month ago
2
Fix op info test for linalg.inv and linalg.inv_ex
#8239
dvhg
closed
1 month ago
0
Previous
Next