DL4J

访问官方
开源的使用JVM部署和训练深度学习模型的套件
Deeplearning4j是为数不多的以Java虚拟机(JVM)为目标,以Java原生编写的机器学习框架之一。该框架由位于旧金山的一组机器学习开发人员开发,并由初创公司Skymind提供商业支持。Deeplearning4j于2017年10月捐赠给了Eclipse基金会。该库与Clojure和Scala兼容。 对于集群和分布式训练,Deeplearning4j与Apache Spark和Apache Hadoop集成。它还与NVIDIA CUDA运行时集成,可在多个GPU之间执行GPU操作和分布式训练。 Deeplearning4j包括一个使用ND4J的n维数组类,该类允许在Java和Scala中进行科学计算,与NumPy提供给Python的函数类似。它可以有效地用作执行线性代数和矩阵操作的库,用于训练和推理。 Deeplearning4j可以用于训练模型,这些模型可以执行图像分类、对象检测、图像分割、自然语言处理和时间序列预测。