NeMo icon indicating copy to clipboard operation
NeMo copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA

NEMO on XLA

Open dennj opened this issue 1 year ago • 0 comments

We need to train NEMO models on specialised hardware XLA/Trainium. Are you planning to make this framework XLA compatible?

dennj avatar Feb 13 '24 11:02 dennj