xformer

A practical implementation of the Transformer neural network architecture.


License
BSD-3-Clause
Install
pip install xformer==1.0.1

Documentation

xformer

This is a lightweight implementation of the decoder from the Transformer architecture. It is implemented both as a direct function on [batch x timesteps x N] sequences, and as an RNNCell.