dm-haiku
immutabledict
jax
jaxlib
msgpack
optax
tensorflow-federated
