AlphaFold v2.2.4

Version v2.2.4 is a bug fix release

Change log

• Bump versions of third party libraries: jax 0.3.17, absl-py 1.0.0, haiku 0.0.7, numpy 1.21.6, tensorflow 2.9.0
• Adapt jnp.take to account for behaviour with the new jax version, see #513 (thanks @sokrypton).
• Reduce size of docker image by removing package caches, see #526 (thanks @TheDen).
• Fix incorrect argument in backbone_loss, see #570 (thanks @sokrypton).