deeplearning4j

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...

AI 简介

Deeplearning4j 是一个基于 JVM 的深度学习模型部署和训练工具套件。它支持从 Keras、TensorFlow 和 ONNX/PyTorch 导入模型，并提供了一个模块化的 C++ 库用于执行数学运算，以及基于该 C++ 库的 Java 数学库。此外，它还包含了 SameDiff 框架，类似于 PyTorch 和 TensorFlow，可以运行深度学习任务。该项目的核心功能包括高阶 API 用于构建多层网络和计算图，支持自定义层和分布式训练，以及强大的线性代数库 ND4J，支持 CPU 和 GPU 加速。Deeplearning4j 适用于需要在 JVM 环境下进行深度学习模型开发和部署的各种场景，特别是那些使用 Java、Scala 或 Kotlin 等语言的项目。

Java

Apache License 2.0

artificial-intelligence clojure deeplearning deeplearning4j dl4j gpu hadoop intellij java linear-algebra matrix-library neural-nets python scala spark

在 GitHub 查看官方网站

14.2k

Stars

3.8k

Forks

747

Watchers

608

Issues

Star 增长

今日0

近 7 天+3

近 30 天+16

综合评分73.1

默认分支master