Model implementation for the contextual embeddings project - View it on GitHub