(Initial version supported CPUs. Later versions added support for GPUs and TPUs).

→ View original post on X — @jeffdean