FlaxNLP Providing flax modules for natural language processing Installation pip install git+https://github.com/altescy/flaxnlp