4 个版本 (重大更改)
0.4.0 | 2024 年 4 月 14 日 |
---|---|
0.3.0 | 2024 年 3 月 31 日 |
0.2.0 | 2024 年 3 月 26 日 |
0.1.0 | 2024 年 1 月 5 日 |
#745 在 硬件支持
每月下载量 237
280KB
2.5K SLoC
Rust Lighter
这个项目最初是我的 RUST 练习,旨在抽象 Rust 最小化 ML 框架 Candle (https://github.com/huggingface/candle) 并引入一种更方便的编程神经网络机器学习模型的方法。
其行为灵感来自 Python KERAS (https://keras.org.cn),初始步骤基于 Rust-Keras-like 代码 (https://github.com/AhmedBoin/Rust-Keras-Like)。
所以我们将项目命名为 Candle Lighter 🕯,因为它可以帮助点亮蜡烛,并且实现起来更加简单。
示例可以在 lib/examples/ 目录下方找到。
将其用作库只需调用 'cargo add candlelighter'
欢迎贡献者
注意:它目前远未达到生产就绪状态,仅用于个人训练目的。不提供任何保证和责任。我是一个个人,不追求任何商业利益。
支持的层类型
元层 | 类型 | 状态 | 示例 |
---|---|---|---|
顺序模型 | - | ✅ | |
- | 特征缩放 | 🏃 | DNN 和 TNN |
- | 密集型 | ✅ | DNN |
- | 卷积 | ✅ | CNN |
- | 池化 | ✅ | - |
- | 归一化 | ✅ | - |
- | 展平 | ✅ | - |
- | 循环 | ✅ | RNN 第一次抛出 |
- | 规范 | ✅ | - |
- | 特征嵌入 | ✅ | S2S 第一次抛出 |
- | 注意力 | 🏃 | TNN 第一次抛出 |
- | 专家混合 | 🏃 | ENN 第一次抛出 |
- | 特征掩码和量化 | 🏃 | - |
并行模型(从分割的角度来看) | - | 🏃 | PNN 第一次抛出 |
并行模型 | 合并 | 🏃 | PNN 第一次抛出 |
- | 模型微调 | 🏃 | - |
许可证
三重许可以兼容 Rust 项目和源根。
根据您的要求,在 MPL 2.0、MIT 许可证 或 Apache 许可证,版本 2.0 下许可。
依赖项
~14–23MB
~355K SLoC