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

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.