!pip install torchinfo
!pip install super_gradients