XLA Pod Request
opooladz opened this issue · comments
Omead Pooladzandi commented
Thank you for the repo.
I am wondering if a recipe for TPU pods can be added. I have access to v4-32 and want to train a LLaMA model from scratch. Wondering if the repo can be extended for this use case.
Thanks
Carlos Mocholí commented
We have support for a limited set of scripts at https://github.com/Lightning-AI/litgpt/tree/main/xla. Give it a shot, it should work with v4-32. Some info may be outdated
Omead Pooladzandi commented
I mostly see fine-tuning code in those scripts. But there seems to be a general lack of pytorch xla examples which are full from scratch solutions.