New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[TPU] Support PyTorch/XLA FSDP via SPMD #28949

Merged

amyeroberts merged 10 commits into huggingface:main from alanwaketan:fsdpv2

Feb 14, 2024

+95 −7

Merged

[TPU] Support PyTorch/XLA FSDP via SPMD #28949

Re-enable checkpointing

CircleCI Checks / run_tests succeeded Feb 14, 2024 in 11m 44s

Workflow: run_tests

View CircleCI workflow

tests_pipelines_torch - Success
tests_flax - Success
tests_torch - Success
tests_custom_tokenizers - Success
examples_tensorflow - Success
tests_exotic_models - Success
tests_pr_documentation_tests - Success
tests_torch_and_tf - Success
tests_tf - Success
examples_torch - Success
tests_hub - Success
tests_onnx - Success
examples_flax - Success
tests_torch_and_flax - Success
tests_pipelines_tf - Success

View more details on CircleCI Checks