Rust在AI方面的应用框架

Burn(Burn)

完整的训练、推理框架。支持cpu、gpu、candle等后端。

支持onnx、pytorch模型导入和运行。好处是可以减少对onnx runtime的依赖

Candle

GitHub - huggingface/candle: Minimalist ML framework for Rust

主要面向推理。支持很多大模型,例如SD等。

Onnx runtime for Rust(推荐前期验证使用)

ultralytics/examples/YOLOv8-ONNXRuntime-Rust at main · ultralytics/ultralytics · GitHub

可以很方便实现目标检测、姿势识别等功能。

在RK3588上完美运行yolov8n-pose(官方原版模型),准确度很高。cpu消耗不大(占用200%, 总cpu 800%)。比基于rknn修改的模型的准确度要好些(可能是模型训练不够所致)。该模型请参考: GitHub - cqu20160901/yolov8pose_rknn_Cplusplus: yolov8pose 瑞芯微 rknn 板端 C++部署。

Mistral

GitHub - EricLBuehler/mistral.rs: Blazingly fast LLM inference.

快速的大语言模型推理平台,提供与OpenAI API兼容的Web服务

Paddle-lite

GitHub - PaddlePaddle/Paddle-Lite: PaddlePaddle High Performance Deep Learning Inference Engine for Mobile and Edge (飞桨高性能深度学习端侧推理引擎)

百度出品的运行时。crates.io有对应的binding。https://crates.io/crates/paddle_inference

未实际使用过。参考示例:

GitHub - PaddlePaddle/Paddle-Lite-Demo: lib, demo, model, data

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值