I am currently working on a school project that consists on analyzing the performance of Machine Learning models in order to assess their scalability and spot potential bottlenecks in their framework.
I am looking for a performance analysis tool that works for Python programs written using Tensorflow's library. My system is running 4 CPU's AMD EPYC 7451 on "Buster" version of Debian. I am willing to visualize the execution trace in order to understand the program's performance.