1. f-string printing with ‘=’:
    • No need to write var name & value separately 
       log_prob = -3.141
 print(f"{log_prob=}")  # log_prob= -3.141
  2. plt.imshow(your_2d_matrix)