TPU support - Githubissues

lucidrains / routing-transformer

Fully featured implementation of Routing Transformer

MIT License

282 stars 29 forks source link

Open abcp4 opened 3 years ago

abcp4 commented 3 years ago

Hello, reading the paper there is some experiment made with TPU v3. I would like to know if there is some code available that supports tpu.