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).