Deeplearning4j是為數(shù)不多的以Java虛擬機(JVM)為目標,以Java原生編寫的機器學習框架之一。該框架由位于舊金山的一組機器學習開發(fā)人員開發(fā),并由初創(chuàng)公司Skymind提供商業(yè)支持。Deeplearning4j于2017年10月捐贈給了Eclipse基金會。該庫與Clojure和Scala兼容。
對于集群和分布式訓練,Deeplearning4j與Apache Spark和Apache Hadoop集成。它還與NVIDIA CUDA運行時集成,可在多個GPU之間執(zhí)行GPU操作和分布式訓練。
Deeplearning4j包括一個使用ND4J的n維數(shù)組類,該類允許在Java和Scala中進行科學計算,與NumPy提供給Python的函數(shù)類似。它可以有效地用作執(zhí)行線性代數(shù)和矩陣操作的庫,用于訓練和推理。
Deeplearning4j可以用于訓練模型,這些模型可以執(zhí)行圖像分類、對象檢測、圖像分割、自然語言處理和時間序列預測。