t training on TPUs and GPUs, but it doesn’t support gradient checkpointing or xFormers. With the same configuration and setup as PyTorch, the Flax training script should be at least ~70% faster! This guide will explore the textual_inversion.py script to help you become more familiar with it, and how you can adapt it for your own use-case. Before running the script, make sure you install the library from source:   Copied git clone https://github