← all repositories

dromara/Omega-AI

A Java deep learning framework for building, training, and serving neural networks with CUDA/CUDNN GPU acceleration.

Omega-AI
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

Omega-AI is a Java-based deep learning framework that enables rapid construction of neural networks including BP networks, CNNs, RNNs, VGG16, ResNet, YOLO, LSTM, Transformer, GPT, LLaMA, and Diffusion models. The engine supports automatic differentiation, multi-threaded computation, and GPU acceleration through JCuda bindings to CUDA and CUDNN. All core algorithms and model implementations are built without third-party ML dependencies, relying solely on CUDA-native libraries.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.