George Necula

I am working on an alternative implementation of host_callback for cpu/gpu using customcall.

BTW, `host_callback` uses the `CustomCall` implementation already by default on CPU (no `outfeed`). There is a new mechanism being developed to replace `host_callback`, but it is not yet ready. In...

@keren9898 I am curious, what is your interest here? Are you associated with any of these libraries?

As Peter said, adding also a question about what APIs are you using directly from shape_poly? It was meant to be exposed directly from `jax2tf` and not from `jax.experimental.export`. In...

Understood. We have fixed already the uses in tensorflowjs internally a few days ago, but they have not released a new version.

@pyu10055 Is there an upcoming tensorflowjs release?