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.