Leavesfly/TinyAI

全栈式轻量级AI框架,TinyAI IS ALL YOU NEED。

What it solves

TinyAI 是一个完全使用 Java 构建的全栈 AI 框架,旨在消除对 Python 和复杂的深度学习环境的依赖。它提供了一个全面的技术栈,弥合了底层数值计算与高层 AI 应用之间的差距,使其既适合教学用途,也适合企业级 Java 集成。

How it works

TinyAI 使用六层架构来组织其 23 个核心模块:

  1. Base Layer: 一个用于张量运算的自定义多维数组库。
  2. Engine Layer: 具有动态计算图的自动微分引擎。
  3. Framework Layer: 核心机器学习工具、神经网络层 (CNN, RNN, Transformer) 以及强化学习算法。
  4. Embodied Intelligence Layer: 用于自动驾驶模拟、机器人控制和世界模型的工具。
  5. Model Layer: LLM 的实现,包括 GPT 系列、DeepSeek R1/V3 以及 Banana 等多模态模型。
  6. Agent Layer: 用于 RAG、多 Agent 协作和认知模式 (ReAct, CoT) 的框架。

Who it’s for

  • Java Developers: 想在不切换到 Python 的情况下将 AI 集成到 Java 生态系统中的开发者。
  • AI Students: 寻求透明、文档齐全 (300k+ 字) 的框架,以从零开始理解 AI 的开发者。
  • Enterprise Architects: 需要具有生产级可靠性和 DevOps 支持的模块化、可扩展 AI 系统架构师。

Highlights

  • Pure Java Implementation: 零 Python 依赖,仅需要 JDK 17+。
  • Broad Scope: 涵盖从基础张量到先进的 DeepSeek R1/V3 推理模型和 VLA (Vision-Language-Action) 架构。
  • Advanced AI Features: 支持 Mixture of Experts (MoE)、RAG 和用于具身智能的 "imagination training" 以提高样本效率。
  • Educational Depth: 包含超过 890 个测试案例,并提供具有丰富文档的结构化学习路径。

相关

  • 项目
  • 项目
  • 项目
  • 项目