当前位置: 首页 > 科技观察

不到1000行代码GitHub千星,天才黑客开源深度学习框架

时间:2023-03-16 12:56:00 科技观察

近日,天才黑客GeorgeHotz开源了一个小型深度学习框架tinygrad,具有PyTorch和micrograd的功能。tinygrad不到1000行代码,目前该项目在GitHub上有1400颗星。在深度学习时代,谷歌、Facebook、百度等科技巨头开源了多种框架,帮助开发者更轻松地学习、构建和训练不同类型的神经网络。而这些大公司也花费了大量的精力去维护TensorFlow、PyTorch等庞大的深度学习框架。除了这些主流框架,开发者还会开源一些小而精的框架或库。例如,今年4月,特斯拉人工智能部门负责人AndrejKarpathy开源了他的micro-autogradenginemicrograd,该引擎还实现了一个50行代码的PyTorchapi-like神经网络库。目前,micrograd项目的GitHubstar已经达到1200颗star。前不久,天才黑客乔治·霍茨(GeorgeHotz)开源了一个小型的AutogradTensor库tinygrad,它介于PyTorch和micrograd之间,可以满足深度学习的大部分需求。上线不到一个月,该项目就已经在GitHub上获得了1400颗星。项目地址:https://github.com/geohot/tinygrad根据GitHub的内容,下面简单介绍一下tinygrad的安装和使用。感兴趣的同学也可以通过GeorgeHotz的YouTube视频进行学习。视频地址:https://www.youtube.com/channel/UCwgKmJM4ZJQRJ-U5NjvR2dgtinygrad安装与使用《tinygrad可能不是最好的深度学习框架,但确实是一个深度学习框架》。George在项目中保证tinygrad代码量永远小于1000行。安装tinygrad的安装过程很简单,只需要使用以下命令:autograd张量库可以满足你对神经网络90%的需求。从tinygrad.optim添加一个优化器(SGD、RMSprop、Adam),并编写一些小批量样板代码来实现您的需要。示例如下:GPUtinygrad通过PyOpenCL支持GPU。但是反向传播暂时不能支持所有的操作。ImageNet推断“麻雀虽小,五脏俱全”。tinygrad还可以支持完整的EfficientNet,输入图像并获取其类别。如果你安装了网络摄像头和cv2,你可以使用下面的代码:注意:如果你想加速,设置GPU=1。测试运行以下代码执行测试:此外,GeorgeHotz还计划加入语言模型、检测模型,进一步减少代码量,提高速度等。天才黑客GeorgeHotz项目的创建者是著名的黑客GeorgeHotz,别名Geohot。1989年出生于美国新泽西州,就读于卡内基梅隆大学罗切斯特理工学院和计算机科学专业生物工程专业。然而,GeorgeHotz在其LinkedIn主页上的教育经历描述中写道:在罗彻斯特理工学院学习期间,他的社团活动是“在宿舍里破解iPhone”……这是他著名的黑客经历之一。2007年,17岁的GeorgeHotz成功破解了iPhone,使手机不再局限于AT&T网络,而是支持其他GSM网络。2009年开发iOS3.1.2越狱软件blackra1n。2010年,GeorgeHotz宣布不再开发越狱软件。然而,他的黑客活动并没有停止。自2009年以来,GeorgeHotz一直在破解PlayStation3(PS3)。2010年初,他宣布拥有对PS3系统内存的读写权限以及对处理器的高级控制权。2011年3月,GeorgeHotz被索尼起诉并和解。此外,GeorgeHotz还破解了三星手机等产品。工作经历方面,曾在谷歌、Facebook、SpaceX工作过。目前,他在他创立的自动驾驶公司comma.ai工作。2015年,GeorgeHotz创立了人工智能创业公司comma.ai,旨在构建基于机器学习算法的自动驾驶技术。GeorgeHotz开发了自动驾驶套件CommaOne,用户只需1000美元就可以将传统汽车升级为自动驾驶版。不过,这个计划后来被取消了。2020年CES大会上,comma.ai展出了其最新产品——comma2,一款驾驶辅助设备,售价999美元。