it's based on a short story which is available online, probably as Fair Use for educational purpose:

https://fyp.uoregon.edu/sites/default/files/chiang_story_of_your_life.pdf

I strongly recommend reading it, as it conveys much more interesting ideas than what was only hinted by the movie.