deepjavalibrary/djl
A Java framework for deep learning that provides engine-agnostic APIs for training and deploying neural network models.

Deep Java Library (DJL) is an open-source, high-level deep learning framework designed for Java developers. It abstracts away the complexity of underlying ML engines (PyTorch, TensorFlow, MXNet, ONNXRuntime) while providing a native Java API for building, training, and deploying neural network models. The framework supports automatic CPU/GPU detection and offers an ergonomic API designed to guide users toward best practices for deep learning tasks.