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 でディープラーニングモデルを構築、訓練、テストしたい 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.