Adrian Wälchli
|
d9493545cf
|
Allow accessing rank information before processes are launched in XLA (#18194)
|
2023-07-31 10:37:35 -04:00 |
|
Carlos Mocholí
|
e9c42ed11f
|
More XLA fixes for nightly support (#18085)
|
2023-07-15 01:16:42 +02:00 |
|
Carlos Mocholí
|
6ec9a6bd9e
|
[TPU] Rename classes to use XLA instead of TPU (#17383)
|
2023-04-28 12:36:22 -04:00 |
|
Carlos Mocholí
|
8dac251273
|
[TPU] Fix PjRT tests (#17408)
|
2023-04-19 16:39:00 +02:00 |
|
Gerson Kroiz
|
7b8fd85e01
|
[TPU] Remove error check for IterableDatasets (#17331)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
|
2023-04-11 22:04:17 +00:00 |
|
Carlos Mocholí
|
ec4f592ecf
|
Sequential `CombinedLoader` to flatten the eval and predict loops (#16726)
|
2023-02-17 17:37:11 +01:00 |
|
Adrian Wälchli
|
91e692c767
|
Rename the TPUSpawnStrategy to XLAStrategy (#16781)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2023-02-17 02:06:24 +00:00 |