tensorflow / tensorflow/tensorflow
tensorflow pjrt plugin
Open
Nobody has claimed this yet.
comp:model
comp:xla
stat:awaiting tensorflower
type:feature
- Dominant language
- C++
- Stars
- 200k
- Forks
- 76.9k
- Avg merge
- 2d 3h
- Merged PRs (30d)
- 433
Description
I've developed a pjrt plugin using https://openxla.org/xla/pjrt_integration as a resource, which is working with jax.
How do I load this plugin and run tensorflow models? Is there some kind of registration step, similar to jax? If I load the plugin into import jax._src.xla_bridge and call tf.config.list_physical_devices() I don't see my device, so I imagine it's something else.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.