Google ने TorchTPU को open-source किया
मूल शीर्षक: Google open-sources TorchTPU, a native PyTorch back end for TPUs
यह क्यों महत्वपूर्ण है
यह PyTorch users को Google TPUs तक सीधी पहुँच देता है, जिससे AI training का hardware विकल्प व्यापक होता है।
Google ने TPU के लिए native PyTorch backend TorchTPU को GitHub पर public कर दिया। यह project PyTorch models को TPU पर चलाने देता है और इसमें custom ATen kernels व compilation cache शामिल हैं।
Google ने अपना PyTorch-TPU integration project TorchTPU open-source कर दिया है। यह repository अब GitHub पर `google-pytorch/torch_tpu` के नाम से उपलब्ध है और इसे अब तक 398 stars मिले हैं। project का मुख्य उद्देश्य PyTorch को Google के Tensor Processing Units (TPUs) पर natively चलाने की सुविधा देना है — बिना किसी भारी workaround के।
तकनीकी रूप से इसमें विभिन्न PyTorch operations के लिए custom ATen kernels लिखे गए हैं। साथ ही एक compilation cache भी है जो बार-बार compile करने की ज़रूरत को कम करता है और execution को तेज़ बनाता है। Distributed training और model examples भी repository में दिए गए हैं।
About सेक्शन में documentation का link `torchtpu.readthedocs.io` पर दिया गया है। Repository में अभी 48 open issues और 221 pull requests हैं, जो दर्शाते हैं कि यह project सक्रिय विकास में है। Build system के रूप में Bazel का उपयोग किया गया है और Docker support भी मौजूद है।
यह कदम उन researchers और developers के लिए अहम है जो PyTorch का उपयोग करते हैं लेकिन Google के TPU hardware पर काम करना चाहते हैं। अब तक TPU पर PyTorch चलाने के लिए `torch_xla` जैसे अप्रत्यक्ष रास्तों की ज़रूरत होती थी।