Instructions seem out of date. There is no requirements.txt. `make html` doesn't work https://github.com/google/flax/tree/main/docs