Need assistance with neural network architectures like transformers?