---
title: XLACheckpointIO
description: >-
  CheckpointIO that utilizes xm.save to save checkpoints for TPU training
  strategies.
canonical: >-
  https://lightning.ai/docs/fabric/stable/full-api-reference/api/generated/lightning.fabric.plugins.io.xla.XLACheckpointIO
---
# XLACheckpointIO {#xlacheckpointio}

### `class lightning.fabric.plugins.io.xla.XLACheckpointIO(*args, **kwargs)` {#lightning.fabric.plugins.io.xla.XLACheckpointIO}

Bases: [`TorchCheckpointIO`](/docs/fabric/2.6.6/full-api-reference/api/generated/lightning.fabric.plugins.io.torch_io.TorchCheckpointIO.md#lightning.fabric.plugins.io.torch_io.TorchCheckpointIO "lightning.fabric.plugins.io.torch_io.TorchCheckpointIO")

CheckpointIO that utilizes `xm.save` to save checkpoints for TPU training strategies.

Warning

This is an [experimental](/docs/pytorch/stable/versioning#experimental-api "(in PyTorch Lightning v2.7.0.dev)") feature.

#### `save_checkpoint(checkpoint, path, storage_options=None)` {#lightning.fabric.plugins.io.xla.XLACheckpointIO.save_checkpoint}

Save model/training states as a checkpoint file through state-dump and file-write.

Parameters:

*   **checkpoint** ([`dict`](https://docs.python.org/3/builtins/stdtypes.html#dict "(in Python v3.15)")\[[`str`](https://docs.python.org/3/builtins/stdtypes.html#str "(in Python v3.15)"), [`Any`](https://docs.python.org/3/library/typing.html#typing.Any "(in Python v3.15)")\]) – dict containing model and trainer state
    
*   **path** (`Union`\[[`str`](https://docs.python.org/3/builtins/stdtypes.html#str "(in Python v3.15)"), [`Path`](https://docs.python.org/3/library/pathlib.html#pathlib.Path "(in Python v3.15)")\]) – write-target path
    
*   **storage\_options** ([`Optional`](https://docs.python.org/3/library/typing.html#typing.Optional "(in Python v3.15)")\[[`Any`](https://docs.python.org/3/library/typing.html#typing.Any "(in Python v3.15)")\]) – not used in `XLACheckpointIO.save_checkpoint`
    

Raises:

[**TypeError**](https://docs.python.org/3/builtins/exceptions.html#TypeError "(in Python v3.15)") – If `storage_options` arg is passed in

Return type:

[`None`](https://docs.python.org/3/builtins/constants.html#None "(in Python v3.15)")

---

**Navigation**

- Parent: [Generated](/docs/fabric/2.6.6/full-api-reference/api/generated.md)