dromara/Omega-AI

Omega-AI：基于java打造的深度学习框架，帮助你快速搭建神经网络，实现模型推理与训练，引擎支持自动求导，多线程与GPU运算，GPU支持CUDA，CUDNN。

What it solves

Omega-AI 是一個完全以 Java 建置的深度學習框架。它旨在讓 Java 開發者更容易進行 AI 開發，並透過從頭實作演算法與模型的方式，深入了解其背後原理。

How it works

此框架提供完整的神經網路建構模組，包括各種層類型（全連接層、卷積層、RNN、LSTM、Transformer 與 Yolo 層）、激活函數（ReLU、Softmax、Sigmoid）以及優化器（Adam、SGD、RMSProp）。透過 JCuda 函式庫，支援 CUDA 與 cuDNN 的 GPU 加速，並可進行多 GPU 訓練。引擎的核心模型與演算法實作不依賴第三方 API。

Who it’s for

想要在不依賴 Python 框架的情況下，使用 Java 建置、訓練與測試深度學習模型的開發者，以及對 AI 模型內部機制有興趣的學生或研究人員。

Highlights

Broad Model Support: Supports a wide range of architectures including BP, CNN, RNN, VGG16, ResNet, YOLO, LSTM, Transformer, GPT, Llama, and Diffusion/Stable Diffusion.
GPU Acceleration: Integrated support for CUDA and cuDNN for high-performance training and inference.
Comprehensive Tooling: Includes built-in data loaders for .bin, .txt, and .idx3-ubyte formats, and various learning rate update strategies.
End-to-End Demos: Provides practical examples ranging from MNIST digit recognition and YOLO object detection to GPT-based chatbots and Stable Diffusion text-to-image generation.