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

..

internal

v4.39 deprecations 🧼 (#29492)

2024-03-07 10:44:43 +00:00

main_classes

Generate: inner decoding methods are no longer public (#29437)

2024-03-05 10:27:36 +00:00

model_doc

Update all references to canonical models (#29001)

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

tasks

Update all references to canonical models (#29001)

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

_toctree.yml

Add models from deit (#28302)

2024-02-01 09:15:55 -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_tensorflow_model.md

Update all references to canonical models (#29001)

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

attention.md

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

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

autoclass_tutorial.md

Update all references to canonical models (#29001)

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

benchmarks.md

Update all references to canonical models (#29001)

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

bertology.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

big_models.md

Update all references to canonical models (#29001)

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

chat_templating.md

[Docs] Add language identifiers to fenced code blocks (#28955)

2024-02-12 10:48:31 -08:00

community.md

Update all references to canonical models (#29001)

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

create_a_model.md

Update all references to canonical models (#29001)

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

custom_models.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

custom_tools.md

Update all references to canonical models (#29001)

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

fast_tokenizers.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

generation_strategies.md

Update all references to canonical models (#29001)

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

glossary.md

Update all references to canonical models (#29001)

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

hpo_train.md

Remove-auth-token (#27060)

2023-11-13 14:20:54 +01: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

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

model_memory_anatomy.md

Update all references to canonical models (#29001)

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

model_sharing.md

Update all references to canonical models (#29001)

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

model_summary.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

multilingual.md

Update all references to canonical models (#29001)

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

pad_truncation.md

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

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

peft.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

perf_hardware.md

Update all references to canonical models (#29001)

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

perf_infer_cpu.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

perf_infer_gpu_many.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

perf_infer_gpu_one.md

F.scaled_dot_product_attention support (#26572)

2023-12-09 05:38:14 +09:00

perf_infer_special.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

perf_torch_compile.md

[Docs] Add language identifiers to fenced code blocks (#28955)

2024-02-12 10:48:31 -08:00

perf_train_cpu_many.md

Update all references to canonical models (#29001)

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

perf_train_cpu.md

Update all references to canonical models (#29001)

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

perf_train_gpu_many.md

Update all references to canonical models (#29001)

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

perf_train_gpu_one.md

Update all references to canonical models (#29001)

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

perf_train_special.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

perf_train_tpu_tf.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

perf_train_tpu.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

performance.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

perplexity.md

Update all references to canonical models (#29001)

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

philosophy.md

[docs] fixed links with 404 (#27327)

2023-11-06 19:45:03 +00:00

pipeline_tutorial.md

Update all references to canonical models (#29001)

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

pipeline_webserver.md

Update all references to canonical models (#29001)

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

pr_checks.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -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

serialization.md

Update all references to canonical models (#29001)

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

task_summary.md

Update all references to canonical models (#29001)

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

tasks_explained.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

testing.md

Make torch xla available on GPU (#29334)

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

tf_xla.md

Update all references to canonical models (#29001)

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

tflite.md

Update all references to canonical models (#29001)

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

tokenizer_summary.md

Update all references to canonical models (#29001)

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

torchscript.md

Update all references to canonical models (#29001)

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

training.md

Update all references to canonical models (#29001)

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

transformers_agents.md

add japanese documentation (#26138)

2023-10-11 10:26:37 -07:00

troubleshooting.md

Update all references to canonical models (#29001)

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