flexformer (pypi): A general implementation of Transformer, to play around with attention and build custom architectures
Keywords: artificial|intelligence|deep|learning|transformer|nlp