An implementation of the Prism layer (https://arxiv.org/abs/2011.04823) - View it on GitHub