Running Captum on Cloud TPUs with TorchTPU: a work log
This is the log of a few days spent running Captum's attribution methods on Cloud TPUs through TorchTPU, in the order the work happened, with what I learned at each step. Unless stated otherwise, measurements are from a Cloud TPU v6e-1 in us-east5-b with torch 2.15.0, torch_tpu 0.2.0, Captum 0.9.0…
Read the full story at DEV Community — Machine Learning ↗
Timeline · 1 report
- 2026-11-01 19:42 · DEV Community — Machine Learning
Running Captum on Cloud TPUs with TorchTPU: a work log