Yitong Huang 873d9bb3cc Make torch xla available on GPU (#29334) ...

* add USE_TORCH_XLA env

* rename torch_tpu to torch_xla

* better is_torch_xla_available; fix some fsdp and performance issues

* fix format

* fix bug when pjrt_device is cpu

* fix bug

* fix the deprecation handling

---------

Co-authored-by: anw90 <ang868@gmail.com>
Co-authored-by: wangang.wa <wangang.wa@alibaba-inc.com>

2024-03-11 14:07:16 +00:00

..

_config.py

german docs translation (#18544)

2022-08-11 09:52:27 -04:00

_toctree.yml

[i18n-de] Translate CONTRIBUTING.md to German (#28954)

2024-02-12 13:39:20 -08:00

accelerate.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

add_new_model.md

[Docs] Fix broken links and syntax issues (#28918)

2024-02-08 14:13:35 -08:00

add_new_pipeline.md

[Docs] Fix backticks in inline code and documentation links (#28875)

2024-02-06 11:15:44 -08:00

add_tensorflow_model.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

autoclass_tutorial.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

contributing.md

[i18n-de] Translate CONTRIBUTING.md to German (#28954)

2024-02-12 13:39:20 -08:00

index.md

[Docs] Add missing language options and fix broken links (#28852)

2024-02-06 12:01:01 -08:00

installation.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

llm_tutorial.md

[Docs] Add missing language options and fix broken links (#28852)

2024-02-06 12:01:01 -08:00

model_sharing.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

peft.md

[i18n-DE] Complete first toc chapter (#26311)

2023-09-27 11:33:05 -07:00

pipeline_tutorial.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

pr_checks.md

[i18n-DE] contribute chapter (#26481)

2023-10-02 09:56:40 -07:00

preprocessing.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

quicktour.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

run_scripts.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

testing.md

Make torch xla available on GPU (#29334)

2024-03-11 14:07:16 +00:00

training.md

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

transformers_agents.md

[i18n-DE] Complete first toc chapter (#26311)

2023-09-27 11:33:05 -07:00