# Resumable JAX training with Orbax

```{literalinclude} ../../examples/example_checkpoint.py
:language: python
```