Raspberry Pi Pico 2W पर चला टाइनी वर्ल्ड मॉडल
मूल शीर्षक: A tiny world model running games on a Raspberry Pi Pico 2 W
यह क्यों महत्वपूर्ण है
Edge AI की सीमाओं को परखता यह प्रोजेक्ट दिखाता है कि बेहद सीमित hardware पर भी world model और RL agent एक साथ चल सकते हैं।
डेवलपर SheepSeb ने दो छोटे neural networks की मदद से Raspberry Pi Pico 2W माइक्रोकंट्रोलर पर गेम generate करने और खेलने वाला 'tinyworldformer' प्रोजेक्ट GitHub पर जारी किया। GitHub पर 159 stars मिले।
SheepSeb नाम के डेवलपर ने 'tinyworldformer' नाम का एक प्रयोगात्मक प्रोजेक्ट बनाया है जो सिर्फ Raspberry Pi Pico 2W जैसे सस्ते माइक्रोकंट्रोलर पर दो छोटे neural networks चलाता है। पहला network 'world model' का काम करता है — यह मौजूदा frame और action देखकर अगला frame और reward predict करता है। दूसरा network 'actor' की भूमिका निभाता है — यह generated frame देखकर अगला action चुनता है।
प्रोजेक्ट की खास बात यह है कि इसमें 'looped transformer' आर्किटेक्चर का उपयोग किया गया है, जहाँ अलग-अलग layers stack करने की बजाय एक ही transformer block को बार-बार same weights के साथ apply किया जाता है। इससे memory की खपत बेहद कम हो जाती है — जो माइक्रोकंट्रोलर के लिए जरूरी है।
Pico 2W को इसलिए चुना गया क्योंकि यह सस्ता है और इसमें FPU (Floating Point Unit) मौजूद है। गेम्स को numpy में सीधे लिखा गया ताकि हर generated frame को real version से verify किया जा सके। बाद में एक streaming RL agent भी जोड़ा गया, ताकि board अपने खुद के 'dream' में game खेलना सीख सके। प्रोजेक्ट Rust में लिखा गया है और GitHub पर अब तक 159 stars मिल चुके हैं।