Omega-AIv1.30.0

基于 Java 打造的 Omega-AI 深度学习框架,支持多种神经网络及模型构建除 cuda 和 cudnn 相关依赖,无其他 API 及第三方依赖包!
 

Omega-AI 核心功能

⚡️ 神经网络

BPNetwork bp神经网络、CNN卷积神经网络、RNN 循环神经网络、LSTM 长短期记忆网络…… 多种神经网络全Java实现

🔑️️ 激活函数

ReluLayer、LeakyReluLayer、TanhLayer、SigmodLayer、SiLULayer……多种激活函数纯手撸

💦️️ 损失函数

MSELoss、CrossEntropyLoss、CrossEntropyLossWithSoftmax、MultiLabelSoftMargin……多种灵活损失函数实现

⛏️ 网络层

Fullylayer全连接层、ConvolutionLayer卷积层、RouteLayer 路由层等…… 提供完善的网络层类型从零到一方案

🔎 归一化

提供BNLayer批归一化、LNLayer 层归一化、InstanceNormaliztionLayer 实例归一化、RMSLayer 均方根归一化

🚀️️ 优化器

内置多种优化器,适配Momentum、Sgd、Adam等场景

️🍃 训练器

实现MBSGDOptimizer训练器,提供给小白直接使用

🍂 学习率更新器

实现CONSTANT、LR_DECAY、GD_GECAY、POLY等学习率覆盖CV到LLM模型

🗳️ 开箱即用

提供SpringBoot从零到一实现的多种深度学习模型案例与模型及权重,真正的开箱即用……

Omega-AI 带领您手把手使用Java熟练掌握深度学习! 点击查看学习思维图(敬请期待)

曾获荣誉


GVP - Gitee 最有价值开源项目


GitCode G-Star 优质开源项目


所在开源社区 “Dromara” 荣获《2024中国互联网发展创新与投资大赛(开源)》二等奖。


Gitee Java深度学习项目推荐榜 Top 1


Dromara 成员AI相关项目

为往圣继绝学,一个人或许能走的更快,但一群人会走的更远。