dm-haiku
immutabledict
jax
jaxlib
msgpack
optax
requests
tensorflow-federated
