Keith Rush

On the more-manual approach: your client / server code looks fine to me--I'd love to see your stacktraces for the `INVALID_ARGUMENT` error. On the docker image: I believe the tutorial...

As far as we can tell, our Docker image seems to be gone--apologies. Filing a bug to create a new release and ensure the image is pushed. On the inability...

It looks to me like we've [deprecated these attributes since our last release](https://github.com/tensorflow/federated/pull/2590). One way we've handled (well, since TFF is developed at HEAD inside of Google, 'handled' is being...

This is definitely possible. The first question I think I might ask is: are you installing a [remote TFF runtime](https://www.tensorflow.org/federated/api_docs/python/tff/backends/native/set_remote_python_execution_context)? By default, TFF will install its local runtime when imported--so...

Apologies, I don't monitor the TFF tags on SO as closely as I once did... I'm not aware of any full-on tutorials we have today. Probably the easiest place to...

+1, ran into this myself

will be in the failing test I'm tagging with this issue, but inlining (imports from pjit_test): ```python mesh = jtu.create_mesh((2, 2), ('x', 'y')) np_inp = np.arange(16).reshape(8, 2) arr = jax.device_put(np_inp,...