Update github CI workflow.
Update github CI workflow.

- use correct branch name
- minimum dependencies are python 3.8 and numpy 1.21.5

PiperOrigin-RevId: 446143942
Change-Id: Ibe3683151004e852343c0b6ba0330665bdefccf6

deepmind/ferminet