coledie/transformer

Transformer layer written in PyTorch.