- 跨平台(硬件:CUDA GPU,
x86
,ARM,PowerPC;操作系统:Windows/Mac/Linux/Android)。
小结一下:模型构建、模型训练和部署一条龙,兼容性强,多线程,跨平台(特别注意Android平台,支持端侧模型)?
3.组件
Deeplearning4j实际上是一堆项目,旨在支持基于 JVM 的深度学习应用程序的所有需求。除了 Deeplearning4j 本身(高级 API),它还包括:
- Deeplearning4j/ScalNet:JVM和Spark上运行神经网络构建、训练和部署的基础框架库;
- ND4J/libND4J:支持CPU/GPU加速的高性能数值计算库,可以说是JVM上的Numpy;
- SameDiff:用于符合微分和计算图库;
- DataVec:数据处理库,提供采样、过滤、变换等操作;
- Arbiter:神经网络超参数搜索和优化库;
- RL4J:JVM上的强化学习库;
- Model Import:模型导入库,可以导入ONNX,TensorFlow,Keras(Caffe)模型;
- Jumpy:ND4J对应Python语言API;
- Python4j:可以在JVM里运行Python脚本语言。
3.1 Deeplearning4j/ScalNet
- 前向神经网络(Feedforward Neural Networks, FNN)
- 自动编码器(AutoEncoders)
- 卷积神经网络(Convolutional Neural Networ