Pytorch library is one of the go-to framework used these days for implementing neural networks or deep learning models. These models have various computational layers which are at times specific to the task & modality(conv1d, conv2d, rnn, lstm, gru, transformer etc.) and at times are generic(batch-norm, dropout, linear etc.). …