Unfinished library for the NLP tasks using Jax and flax
Python
NLP library
I wanted to create a library for the NLP tasks using Jax and flax. However, I didn't finish it properly. Therefore, I decided to share these codes. I hope that these codes could be useful for someone.