Launchpad Release #41
Description
Hi Launchpad Team,
Thanks for trying to help us with the #32, however, I don't think we have a release for the new TF version. Both the nightly and stable versions?
I am currently using a wheel built from a source as mentioned in #30 (comment). There are some additional fixes that seem to be required to work with TF 2.12.0. I am happy to create a PR if that is preferred (but maybe it's easier to update from internal). There seem to be some upgrades needed in terms of how to handle absl Status and tsl Status otherwise things are straightforward.
The update is quite useful as it is currently blocking using JAX agents in dm-acme at HEAD. dm-acme@HEAD now uses jax>0.4.3 which is incompatible with the TF/TFP version used in Acme.
@ddmbr what do you think?