网易首页 > 网易号 > 正文 申请入驻

纯Rust编写的机器学习框架Neuronika,速度堪比PyTorch

0
分享至

机器之心报道

编辑:陈萍

这是一个用 Rust 编写的机器学习框架,与 PyTorch 类似,现已实现最常见的层组件(dense 层、dropout 层等),速度堪比 PyTorch。

Rust 作为一门系统编程语言,专注于安全,尤其是并发安全,支持函数式和命令式以及泛型等编程范式的多范式语言,其语法上与 C++ 类似。

随着 Rust 语言热度的强劲增长,也有许多研究者探索 Rust 在机器学习等方面的研究。近日,有研究者用 Rust 编写了一个机器学习框架 Neuronika。

Neuronika 是由 Francesco Iannelli 等人开发,他们现在是计算机科学的硕士生。该框架提供了自动微分和动态神经网络,非常像 Pytorch。目前已经实现了最常见的层组件,如 dense 层、dropout 层、GRU、LSTM 和 1d-2d-3d CNN,然而,缺少池化层等等。Neuronika 还提供了损失函数、优化器、计算图、张量和数据实用程序。

速度方面,项目作者表示 Neuronika 的表现可与 PyTorch 媲美。你可以对它进行基准测试。但目前还不能支持 GPU。想要进行基准测试,可以参考测试文档。

测试文档地址:https://docs.rs/neuronika/0.1.0/neuronika/

项目地址:https://github.com/neuronika/neuronika

对于该机器学习框架,网友直呼非常棒!

对于还不能支持 GPU,有网友支招:「对于 GPU 支持,可考虑 Vulkan 计算着色器。事实证明,它们的性能非常好,而且几乎所有的 GPU 平台都支持它,包括智能手机和浏览器。」

还有网友表示担心,「如果不支持 GPU,能训练出可用的模型吗?现在关于机器学习的文章,他们都在谈论更大、更昂贵的 GPU。」对于这一担心,项目作者表示:「这取决于用例。当然,GPU 支持是当今必须的,我们认为它是我们框架的开发的一个重大里程碑。」

项目介绍

Neuronika 是用纯 Rust 语言编写的机器学习框架,其构建重点在于易用性、快速原型开发和高效性能。

Neuronika 的核心机制是一种叫做反向模式自动微分机制( reverse-mode automatic differentiation),这种机制允许用户在改变动态神经网络时可以轻松实现,没有任何开销,可通过 API 运行。

Rust 语言提供了一个直观、轻巧且易于使用的界面,同时实现了令人难以置信的性能。

以下 crate 特征 flag 可用,配置 ndarray 后端:

  • serialize:可以序列化支持 serde 1.x。
  • blas:为矩阵乘法提供透明的 BLAS 支持。将 blas-src 用于 pluggable 后端,需要单独配置。
  • matrixmultiply-threading:启用 matrixmultiply 包中的 threading 特性。

目前,Neuronika 的开发处于初始阶段,但其发展迅速。研究者还在不断的开发该项目,从一个版本到另一个版本的转换过程中,预计会有突破性的变化。如果可能的话,研究者会采用最新的、稳定的 Rust 特征来开发。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
《追风者》:“一脸苦相”却要演“女特工”,谁的审美出了问题?

《追风者》:“一脸苦相”却要演“女特工”,谁的审美出了问题?

娱乐圈酸柠檬
2024-03-25 02:43:04
妖怪为啥不趁镇元子外出去抢人参果?网友:你猜清风明月是啥级别

妖怪为啥不趁镇元子外出去抢人参果?网友:你猜清风明月是啥级别

牛锅巴小钒
2024-03-29 07:07:25
雷军不厚道,小米SU7重新定义汽车大定

雷军不厚道,小米SU7重新定义汽车大定

数码圈麻辣烫
2024-03-29 00:18:21
外媒:荷兰首相吕特与北大师生交流,称北约与亚洲关系对保持世界稳定至关重要

外媒:荷兰首相吕特与北大师生交流,称北约与亚洲关系对保持世界稳定至关重要

环球网资讯
2024-03-28 17:31:53
抓错了?俄方抓捕的顶包货?被抓前后差异巨大,IS称须为事件负责

抓错了?俄方抓捕的顶包货?被抓前后差异巨大,IS称须为事件负责

椰青美食分享
2024-03-28 17:26:25
吴越50岁探班,尽显老有肌肤,46岁男神依然魅力无限。

吴越50岁探班,尽显老有肌肤,46岁男神依然魅力无限。

八卦娱乐名
2024-03-28 20:47:21
恐袭事件后,俄做出两大决定,欧洲认为:中国该去找普京谈一谈了

恐袭事件后,俄做出两大决定,欧洲认为:中国该去找普京谈一谈了

千里持剑
2024-03-28 14:23:21
幼儿园变相“摸家底”,让学生和家里车拍照,家长反套路操作绝了

幼儿园变相“摸家底”,让学生和家里车拍照,家长反套路操作绝了

红丽说教育
2024-02-05 09:43:36
越南一队人马,在中国待了一周,内部政局变动后,要和中国对下表

越南一队人马,在中国待了一周,内部政局变动后,要和中国对下表

立马看世界
2024-03-29 10:45:15
真厉害!乒乓球比赛伊藤美诚妈妈突然叫暂停, 居然让女儿连得7分

真厉害!乒乓球比赛伊藤美诚妈妈突然叫暂停, 居然让女儿连得7分

刘哥谈体育
2024-03-29 12:25:48
石油系统内部人说,新能源车续航突破2000公里,加油站很慌

石油系统内部人说,新能源车续航突破2000公里,加油站很慌

资本百科
2024-03-28 14:05:38
金价狂飙再创历史新高!花旗银行:金价或飙升至每盎司3000美元

金价狂飙再创历史新高!花旗银行:金价或飙升至每盎司3000美元

同花顺期货通
2024-03-29 10:30:45
八村塁再砍32+10!哈姆是真要后悔了,早这么干湖人哪至于排第九

八村塁再砍32+10!哈姆是真要后悔了,早这么干湖人哪至于排第九

奕辰说球
2024-03-29 11:39:44
马卡:阿尔维斯已上交护照,且被禁止接近受害人

马卡:阿尔维斯已上交护照,且被禁止接近受害人

懂球帝
2024-03-28 00:06:36
开车时的不雅行为,摄像头拍的一清二楚,网友:不要这么迫不及待

开车时的不雅行为,摄像头拍的一清二楚,网友:不要这么迫不及待

芷夕说文
2024-01-08 08:54:41
马琳为何放弃孙颖莎临场指挥?对阵平野美宇,看他在场边举动秒懂

马琳为何放弃孙颖莎临场指挥?对阵平野美宇,看他在场边举动秒懂

东球弟
2024-03-28 15:45:49
大S太惨!台媒坐实嗑药,汪小菲带走保姆司机,曝具妈让光头离婚

大S太惨!台媒坐实嗑药,汪小菲带走保姆司机,曝具妈让光头离婚

娱乐圈酸柠檬
2024-03-29 01:47:05
国防部都转发了!中国成功发射云海三号02星,到底有什么用?

国防部都转发了!中国成功发射云海三号02星,到底有什么用?

环球科学猫
2024-03-27 16:19:14
3-1!巴西一哥逆转战胜法国新秀,晋级男单八强,对手心态被打崩

3-1!巴西一哥逆转战胜法国新秀,晋级男单八强,对手心态被打崩

湘楚风云
2024-03-29 11:36:23
李铁后院“着火”让他崩溃,龙菲到底什么人,生活照曝光:挺漂亮

李铁后院“着火”让他崩溃,龙菲到底什么人,生活照曝光:挺漂亮

子芫伴你成长
2024-01-14 21:46:43
2024-03-29 13:48:49
机器之心Pro
机器之心Pro
专业的人工智能媒体
8835文章数 141871关注度
往期回顾 全部

科技要闻

雷军:我们是卷王,建议BBA车主感受下时代

头条要闻

赖清德就职前 美众议院军委会成员窜台

头条要闻

赖清德就职前 美众议院军委会成员窜台

体育要闻

拒绝为国出战,他是足坛"天选打工人"

娱乐要闻

胡夏被曝有孩子!工作室火速辟谣

财经要闻

月薪1万!万科高管自愿降薪 曾年薪千万

汽车要闻

找回久违的开怀大笑 试驾小米SU7 Max

态度原创

手机
亲子
旅游
房产
公开课

手机要闻

折叠屏难产 苹果iPhone Fold最快2026年推出

亲子要闻

4个春天长高冠军,矮个娃个子猛冲185

旅游要闻

赏花不止于“春” 农文旅融合助力乡村

房产要闻

卖方“厮杀”,广州楼市掀起特价潮

公开课

30岁之前,你要学会的13件事情

无障碍浏览 进入关怀版