UPMEM LLM Framework allows profiling PyTorch layers and functions and simulate those layers/functions with a given hardware profile. - View it on GitHub