please add a module to demonstrate how to convert NumPy code to run on Trainium (via nki.language) #6
Description
Can we please add a module to demonstrate how to convert NumPy code to run on Trainium (via nki.language)? That would serve a growing number of researchers who would like to use Trainium as a general HPC engine. It would be most natural for those researchers to code in NumPy. From there they need to know how to use nki.language to convert their NumPy code to run on Trainium.