[R] Training Stable Diffusion on TPUs

Has anyone experimented with SD training/finetuning on TPUs, ideally in JAX?

I have found two promising repositories:

- https://github.com/patil-suraj/stable-diffusion-jax
- https://github.com/huggingface/diffusers/issues/475

However none of them looks ready for use. Is there any alternative, or will I have to get coding? ;-)

diffusion machinelearning stable diffusion tpus training