AI 开发框架

DL4J

开源的使用JVM部署和训练深度学习模型的套件

标签:
其他站点:GitHub项目地址

Deeplearning4j是少数几个基于Java虚拟机(JVM)并用原生Java编写的人工智能框架之一。该框架由位于旧金山的一群机器学习专家开发,商业支持则来自初创公司Skymind。在2017年10月,Deeplearning4j被捐赠给了Eclipse基金会。此外,它还与Clojure和Scala兼容。

为了实现集群和分布式训练功能,Deeplearning4j集成了Apache Spark和Apache Hadoop,并且还能通过NVIDIA CUDA运行时支持多GPU操作及分布式的培训任务。

该框架内含ND4J的n维数组类库,使在Java与Scala中进行科学计算变得便捷。这一工具类似于NumPy提供的功能,在执行线性代数运算和矩阵处理方面表现出色,并有助于模型训练与推理过程中的数据管理。

Deeplearning4j支持构建多种类型的深度学习模型,包括图像分类、目标检测、图像分割以及自然语言处理领域的问题解决能力,同时还能进行时间序列预测。

相关导航