NEMO on XLA
This issue is stale because it has been open for 30 days with no activity. Remove stale label or comment or this will be closed in 7 days. This issue was closed because it has been inactive for 7 days since being marked as stale.
We need to train NEMO models on specialised hardware XLA/Trainium.
Are you planning to make this framework XLA compatible?