Style: Fix linting errors and organize imports (ruff & mypy)

chore: snapshot current refinements

Implemented the following parts for the Transformer model: poisional_encoding, feedfoward, encoder, and skeleton of decoder. As well as Test cases for each class respectively and visualizations