dromara / Omega-AI

git clone https://github.com/dromara/Omega-AI.git

Omega-AI is a Java-based deep learning framework that helps you quickly build neural networks for inference and training. Its engine supports automatic differentiation, multithreading, and GPU acceleration with CUDA and cuDNN.