Google Rilis TorchTPU sebagai Backend PyTorch untuk TPU
Judul asli: Google open-sources TorchTPU, a native PyTorch back end for TPUs
Mengapa Ini Penting
Memperluas akses TPU ke ekosistem PyTorch yang dominan di riset AI global.
Google membuka kode TorchTPU di GitHub, backend PyTorch native untuk TPU yang mendukung custom ATen kernel, compilation cache, dan pelatihan model terdistribusi.
Google merilis TorchTPU sebagai proyek open-source di bawah repositori github.com/google-pytorch/torch_tpu. Proyek ini menyediakan backend PyTorch langsung untuk Tensor Processing Unit (TPU) milik Google, memungkinkan pengguna menjalankan model dan operasi PyTorch di perangkat TPU tanpa lapisan abstraksi tambahan.
Fitur utama mencakup custom ATen kernel untuk berbagai operasi PyTorch, compilation cache guna mengoptimalkan eksekusi berulang, serta contoh penggunaan untuk pelatihan terdistribusi. Repositori tercatat memiliki 4.466 commit, 398 bintang, dan 20 fork sejak dipublikasikan. Terdapat 48 isu terbuka dan 221 pull request aktif — menandakan pengembangan yang cukup intensif.
Dokumentasi resmi tersedia di torchtpu.readthedocs.io. Proyek ini juga menyertakan kebijakan keamanan dan panduan kontribusi, menunjukkan kesiapan Google untuk menerima partisipasi komunitas secara terbuka.