TPU Training - Githubissues

OpenNMT / OpenNMT-tf

Neural machine translation and sequence learning using TensorFlow

https://opennmt.net/

MIT License

1.46k stars 392 forks source link

Open tanmayb123 opened 5 years ago

tanmayb123 commented 5 years ago

Does OpenNMT-tf support training on Cloud TPUs?

guillaumekln commented 5 years ago

Currently it does not.

Marking this as "contributions welcome", it would be nice to support this.

ZenBel commented 1 year ago

Is there any update on this since 2019?

guillaumekln commented 1 year ago

We now support XLA compilation which could be a first step towards supporting TPUs.

However, I don't plan to make more developments in this direction. We only use GPUs at the moment.