shell pip install snakeviz python import snakeviz import numpy as np @snakeviz.profile def compute_something(): a = np.random.randn(1000, 1000) b = np.random.randn(1000, 1000) c = np.matmul(a, b) compute_something()